cmpute / pytorch-cmake-exampleLinks
Example to build PyTorch CUDA extension using CMake (with pybind11 and scikit-build)
☆12Updated 5 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:
- ☆33Updated 4 years ago
- Step-by-step optimization of CUDA SGEMM☆390Updated 3 years ago
- Kernel Tuner☆371Updated last week
- A library of GPU kernels for sparse matrix operations.☆275Updated 4 years ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆585Updated this week
- CUDA Matrix Multiplication Optimization☆235Updated last year
- Efficient Distributed GPU Programming for Exascale, an SC/ISC Tutorial☆314Updated last week
- CUTLASS and CuTe Examples☆98Updated 3 weeks ago
- Examples and exercises from the book Programming Massively Parallel Processors - A Hands-on Approach. David B. Kirk and Wen-mei W. Hwu (T…☆75Updated 4 years ago
- Fastest kernels written from scratch☆386Updated last month
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆360Updated this week
- ☆180Updated last year
- The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…☆466Updated last week
- Training material for Nsight developer tools☆170Updated last year
- ☆123Updated 2 weeks ago
- A Quirky Assortment of CuTe Kernels☆651Updated last week
- An extension library of WMMA API (Tensor Core API)☆108Updated last year
- The CUDA target for Numba☆207Updated last week
- A plugin for Jupyter Notebook to run CUDA C/C++ code☆248Updated last year
- Efficient SpGEMM on GPU using CUDA and CSR☆57Updated 2 years ago
- ☆243Updated last year
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆277Updated this week
- NVIDIA NVSHMEM is a parallel programming interface for NVIDIA GPUs based on OpenSHMEM. NVSHMEM can significantly reduce multi-process com…☆369Updated 3 weeks ago
- A simple high performance CUDA GEMM implementation.☆415Updated last year
- Matrix Multiply-Accumulate with CUDA and WMMA( Tensor Core)☆145Updated 5 years ago
- CUDA Kernel Benchmarking Library☆757Updated 2 weeks ago
- ☆590Updated 2 weeks ago
- ☆246Updated this week
- Samples demonstrating how to use the Compute Sanitizer Tools and Public API☆89Updated 2 years ago
- 🤖FFPA: Extend FlashAttention-2 with Split-D, ~O(1) SRAM complexity for large headdim, 1.8x~3x↑🎉 vs SDPA EA.☆227Updated 3 months ago