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
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆640Updated this week
- Kernel Tuner☆372Updated 2 weeks ago
- A library of GPU kernels for sparse matrix operations.☆277Updated 5 years ago
- The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…☆472Updated this week
- Step-by-step optimization of CUDA SGEMM☆405Updated 3 years ago
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆363Updated this week
- ☆184Updated last year
- Training material for Nsight developer tools☆173Updated last year
- CUDA Matrix Multiplication Optimization☆239Updated last year
- CUDA Kernel Benchmarking Library☆769Updated 2 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
- ☆59Updated 2 months ago
- Matrix Multiply-Accumulate with CUDA and WMMA( Tensor Core)☆146Updated 5 years ago
- The Foundation for All Legate Libraries☆232Updated last week
- ☆598Updated last month
- Efficient Distributed GPU Programming for Exascale, an SC/ISC Tutorial☆332Updated 2 weeks ago
- Fastest kernels written from scratch☆400Updated 2 months ago
- An extension library of WMMA API (Tensor Core API)☆109Updated last year
- Examples demonstrating available options to program multiple GPUs in a single node or a cluster☆833Updated 2 months ago
- CUTLASS and CuTe Examples☆104Updated this week
- CUDA by Example, written by two senior members of the CUDA software platform team, shows programmers how to employ this new technology. …☆459Updated 2 years ago
- ☆256Updated this week
- ☆246Updated last year
- CUDA by practice☆130Updated 5 years ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆294Updated this week
- A Quirky Assortment of CuTe Kernels☆675Updated last week
- A library to analyze PyTorch traces.☆437Updated last week
- Samples demonstrating how to use the Compute Sanitizer Tools and Public API☆90Updated 2 years ago
- Efficient SpGEMM on GPU using CUDA and CSR☆57Updated 2 years ago