thu-pacman / PETLinks
PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections
☆122Updated 3 years ago
Alternatives and similar repositories for PET
Users that are interested in PET are comparing it to the libraries listed below
Sorting:
- DietCode Code Release☆64Updated 3 years ago
- [MLSys 2021] IOS: Inter-Operator Scheduler for CNN Acceleration☆200Updated 3 years ago
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆51Updated last year
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆141Updated 2 years ago
- ☆92Updated 3 years ago
- play gemm with tvm☆92Updated 2 years ago
- System for automated integration of deep learning backends.☆47Updated 3 years ago
- A home for the final text of all TVM RFCs.☆108Updated last year
- ☆192Updated 2 years ago
- Automatic Schedule Exploration and Optimization Framework for Tensor Computations☆180Updated 3 years ago
- ☆83Updated 3 years ago
- ☆145Updated 10 months ago
- ☆42Updated 2 years ago
- Automatic Mapping Generation, Verification, and Exploration for ISA-based Spatial Accelerators☆119Updated 3 years ago
- We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …☆190Updated 10 months ago
- Artifacts of EVT ASPLOS'24☆28Updated last year
- ☆41Updated 3 years ago
- ☆41Updated last year
- ☆40Updated 5 years ago
- ☆163Updated last year
- ☆110Updated last year
- ASPLOS'24: Optimal Kernel Orchestration for Tensor Programs with Korch☆39Updated 8 months ago
- Magicube is a high-performance library for quantized sparse matrix operations (SpMM and SDDMM) of deep learning on Tensor Cores.☆90Updated 3 years ago
- Benchmark scripts for TVM☆74Updated 3 years ago
- Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616☆133Updated 2 years ago
- An unofficial cuda assembler, for all generations of SASS, hopefully :)☆84Updated 2 years ago
- ☆23Updated 4 months ago
- A benchmark suited especially for deep learning operators☆42Updated 2 years ago
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆96Updated 3 months ago
- ☆119Updated 8 months ago