thu-pacman / PET
PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections
☆119Updated 2 years ago
Alternatives and similar repositories for PET:
Users that are interested in PET are comparing it to the libraries listed below
- DietCode Code Release☆61Updated 2 years ago
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆51Updated 8 months ago
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆135Updated last year
- ☆75Updated 2 years ago
- Automatic Schedule Exploration and Optimization Framework for Tensor Computations☆176Updated 2 years ago
- A home for the final text of all TVM RFCs.☆103Updated 5 months ago
- System for automated integration of deep learning backends.☆48Updated 2 years ago
- [MLSys 2021] IOS: Inter-Operator Scheduler for CNN Acceleration☆197Updated 2 years ago
- ☆44Updated last year
- ☆92Updated 2 years ago
- Automatic Mapping Generation, Verification, and Exploration for ISA-based Spatial Accelerators☆108Updated 2 years ago
- play gemm with tvm☆89Updated last year
- ☆141Updated last month
- ☆37Updated 2 years ago
- Benchmark scripts for TVM☆73Updated 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 …☆178Updated last month
- ASPLOS'24: Optimal Kernel Orchestration for Tensor Programs with Korch☆32Updated 7 months ago
- An Efficient Pipelined Data Parallel Approach for Training Large Model☆74Updated 4 years ago
- ☆137Updated 8 months ago
- ☆194Updated last year
- Artifacts of EVT ASPLOS'24☆24Updated last year
- ☆91Updated 11 months ago
- ☆87Updated last week
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆89Updated 3 weeks ago
- Repository for SysML19 Artifacts Evaluation☆53Updated 6 years ago
- Magicube is a high-performance library for quantized sparse matrix operations (SpMM and SDDMM) of deep learning on Tensor Cores.☆86Updated 2 years ago
- The quantitative performance comparison among DL compilers on CNN models.☆74Updated 4 years ago
- GVProf: A Value Profiler for GPU-based Clusters☆49Updated 11 months ago
- An unofficial cuda assembler, for all generations of SASS, hopefully :)☆82Updated 2 years ago
- Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616☆133Updated last year