cmpute / pytorch-cmake-exampleLinks
Example to build PyTorch CUDA extension using CMake (with pybind11 and scikit-build)
☆11Updated 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:
- ☆34Updated 4 years ago
- Kernel Tuner☆383Updated this week
- A library of GPU kernels for sparse matrix operations.☆283Updated 5 years ago
- CUDA Matrix Multiplication Optimization☆256Updated last year
- Step-by-step optimization of CUDA SGEMM☆428Updated 3 years ago
- The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…☆507Updated last week
- CUDA Kernel Benchmarking Library☆809Updated this week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆380Updated this week
- Training material for Nsight developer tools☆177Updated last year
- Matrix Multiply-Accumulate with CUDA and WMMA( Tensor Core)☆145Updated 5 years ago
- An extension library of WMMA API (Tensor Core API)☆109Updated last year
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆739Updated this week
- ☆615Updated last week
- The Foundation for All Legate Libraries☆233Updated last week
- This repository contains examples CUDA usage in Cython code.☆26Updated 4 years ago
- C++ library for reading and writing of numpy's .npy files☆424Updated last year
- ☆59Updated 4 months ago
- ☆189Updated last year
- Examples demonstrating available options to program multiple GPUs in a single node or a cluster☆864Updated 4 months ago
- Training materials associated with NVIDIA's CUDA Training Series (www.olcf.ornl.gov/cuda-training-series/)☆934Updated last year
- Some CUDA design patterns and a bit of template magic for CUDA☆158Updated 2 years ago
- Samples demonstrating how to use the Compute Sanitizer Tools and Public API☆93Updated 2 years ago
- Template for GPU accelerated python libraries☆51Updated 2 years ago
- The CUDA target for Numba☆251Updated this week
- The simplest but fast implementation of matrix multiplication in CUDA.☆40Updated last year
- CUDA official sample codes☆371Updated 10 years ago
- A simple high performance CUDA GEMM implementation.☆426Updated 2 years ago
- Helpful kernel tutorials and examples for tile-based GPU programming☆630Updated this week
- Efficient Distributed GPU Programming for Exascale, an SC/ISC Tutorial☆350Updated 2 months ago
- An example combining scikit-build and pybind11☆141Updated last week