cmpute / pytorch-cmake-example
Example to build PyTorch CUDA extension using CMake (with pybind11 and scikit-build)
☆11Updated 4 years ago
Related projects: ⓘ
- ☆24Updated 3 years ago
- This repository contains examples CUDA usage in Cython code.☆19Updated 2 years ago
- NVIDIA Math Libraries for the Python Ecosystem☆194Updated 2 months ago
- Kernel Tuner☆273Updated this week
- A library of GPU kernels for sparse matrix operations.☆240Updated 3 years ago
- Extending JAX with custom C++ and CUDA code☆369Updated last month
- Template for GPU accelerated python libraries☆44Updated last year
- Step-by-step optimization of CUDA SGEMM☆207Updated 2 years ago
- ☆137Updated 3 months ago
- Fusing Taichi into PyTorch☆134Updated last year
- Training materials associated with NVIDIA's CUDA Training Series (www.olcf.ornl.gov/cuda-training-series/)☆545Updated last month
- Matrix Multiply-Accumulate with CUDA and WMMA( Tensor Core)☆109Updated 4 years ago
- ☆124Updated last week
- Customized matrix multiplication kernels☆53Updated 2 years ago
- Examples and exercises from the book Programming Massively Parallel Processors - A Hands-on Approach. David B. Kirk and Wen-mei W. Hwu (T…☆33Updated 3 years ago
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆250Updated this week
- Fast CUDA matrix multiplication from scratch☆423Updated 8 months ago
- GitHub Action to install CUDA☆145Updated 2 months ago
- OpTree: Optimized PyTree Utilities☆136Updated this week
- Some CUDA design patterns and a bit of template magic for CUDA☆144Updated last year
- CUDA Matrix Multiplication Optimization☆118Updated 2 months ago
- An example combining scikit-build and pybind11☆106Updated this week
- Efficient SpGEMM on GPU using CUDA and CSR☆50Updated last year
- ☆138Updated 2 months ago
- A nanobind example project☆87Updated this week
- jax-triton contains integrations between JAX and OpenAI Triton☆328Updated this week
- CUDA Kernel Benchmarking Library☆482Updated 3 months ago
- Code samples for the CUDA tutorial "CUDA and Applications to Task-based Programming"☆78Updated last year
- Training material for Nsight developer tools☆125Updated last month
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆171Updated 5 years ago