sandeepkumar-skb / pytorch_custom_opLinks
End to End steps for adding custom ops in PyTorch.
☆23Updated 5 years ago
Alternatives and similar repositories for pytorch_custom_op
Users that are interested in pytorch_custom_op are comparing it to the libraries listed below
Sorting:
- ☆107Updated last year
- An extension library of WMMA API (Tensor Core API)☆105Updated last year
- ☆116Updated 8 months ago
- Matrix Multiply-Accumulate with CUDA and WMMA( Tensor Core)☆143Updated 5 years ago
- ☆82Updated 2 years ago
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆97Updated 2 months ago
- We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …☆185Updated 7 months ago
- ☆140Updated 4 months ago
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆116Updated 4 months ago
- ☆57Updated 3 months ago
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆94Updated last week
- A tool for examining GPU scheduling behavior.☆88Updated last year
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆114Updated last year
- NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer☆128Updated this week
- Artifacts of EVT ASPLOS'24☆26Updated last year
- ☆233Updated last year
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆51Updated last year
- study of Ampere' Sparse Matmul☆18Updated 4 years ago
- ☆90Updated 10 months ago
- Standalone Flash Attention v2 kernel without libtorch dependency☆110Updated last year
- Magicube is a high-performance library for quantized sparse matrix operations (SpMM and SDDMM) of deep learning on Tensor Cores.☆89Updated 2 years ago
- ☆42Updated 4 months ago
- Github mirror of trition-lang/triton repo.☆75Updated this week
- Dissecting NVIDIA GPU Architecture☆105Updated 3 years ago
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆138Updated 2 years ago
- ☆50Updated 6 years ago
- CUDA Matrix Multiplication Optimization☆222Updated last year
- An experimental CPU backend for Triton☆153Updated 3 months ago
- Benchmark code for the "Online normalizer calculation for softmax" paper☆99Updated 7 years ago
- An experimental CPU backend for Triton (https//github.com/openai/triton)☆45Updated last month