r-barnes / pytorch_cmake_exampleLinks
☆32Updated 4 years ago
Alternatives and similar repositories for pytorch_cmake_example
Users that are interested in pytorch_cmake_example are comparing it to the libraries listed below
Sorting:
- A library of GPU kernels for sparse matrix operations.☆270Updated 4 years ago
- Matrix Multiply-Accumulate with CUDA and WMMA( Tensor Core)☆138Updated 4 years ago
- CUDA Matrix Multiplication Optimization☆214Updated last year
- Efficient SpGEMM on GPU using CUDA and CSR☆57Updated 2 years ago
- Step-by-step optimization of CUDA SGEMM☆363Updated 3 years ago
- ☆106Updated last year
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆223Updated this week
- Training material for Nsight developer tools☆163Updated last year
- Assembler for NVIDIA Volta and Turing GPUs☆226Updated 3 years ago
- CUTLASS and CuTe Examples☆68Updated 3 weeks ago
- An extension library of WMMA API (Tensor Core API)☆99Updated last year
- ☆129Updated 3 months ago
- ☆94Updated 8 years ago
- A simple high performance CUDA GEMM implementation.☆393Updated last year
- ☆229Updated last year
- Kernel Tuner☆357Updated 2 weeks ago
- Some source code about matrix multiplication implementation on CUDA☆34Updated 6 years ago
- Magicube is a high-performance library for quantized sparse matrix operations (SpMM and SDDMM) of deep learning on Tensor Cores.☆89Updated 2 years ago
- Efficient Top-K implementation on the GPU☆183Updated 6 years ago
- Dissecting NVIDIA GPU Architecture☆103Updated 3 years ago
- 🎃 GPU load-balancing library for regular and irregular computations.☆62Updated last year
- We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …☆183Updated 6 months ago
- Examples and exercises from the book Programming Massively Parallel Processors - A Hands-on Approach. David B. Kirk and Wen-mei W. Hwu (T…☆72Updated 4 years ago
- A language and compiler for irregular tensor programs.☆149Updated 8 months ago
- A Easy-to-understand TensorOp Matmul Tutorial☆370Updated 10 months ago
- Optimizing SGEMM kernel functions on NVIDIA GPUs to a close-to-cuBLAS performance.☆371Updated 7 months ago
- Example to build PyTorch CUDA extension using CMake (with pybind11 and scikit-build)☆11Updated 5 years ago
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆93Updated last month
- ☆109Updated 4 years ago
- ☆51Updated 6 years ago