thu-pacman / PET
PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections
☆114Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for PET
- DietCode Code Release☆62Updated 2 years ago
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆49Updated 3 months ago
- Automatic Mapping Generation, Verification, and Exploration for ISA-based Spatial Accelerators☆103Updated 2 years ago
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆131Updated last year
- Automatic Schedule Exploration and Optimization Framework for Tensor Computations☆176Updated 2 years ago
- ☆44Updated last year
- ☆90Updated 2 years ago
- ☆73Updated last year
- play gemm with tvm☆84Updated last year
- A home for the final text of all TVM RFCs.☆101Updated last month
- [MLSys 2021] IOS: Inter-Operator Scheduler for CNN Acceleration☆195Updated 2 years ago
- System for automated integration of deep learning backends.☆48Updated 2 years ago
- ASPLOS'24: Optimal Kernel Orchestration for Tensor Programs with Korch☆29Updated 3 months ago
- Benchmark scripts for TVM☆73Updated 2 years ago
- ☆196Updated last year
- ☆35Updated 2 years ago
- The quantitative performance comparison among DL compilers on CNN models.☆75Updated 4 years ago
- TiledCUDA is a highly efficient kernel template library designed to elevate CUDA C’s level of abstraction for processing tiles.☆154Updated this week
- ☆131Updated 3 months ago
- ☆140Updated last year
- An Efficient Pipelined Data Parallel Approach for Training Large Model☆70Updated 3 years ago
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆75Updated this week
- GVProf: A Value Profiler for GPU-based Clusters☆47Updated 7 months ago
- Magicube is a high-performance library for quantized sparse matrix operations (SpMM and SDDMM) of deep learning on Tensor Cores.☆81Updated last year
- MAGIS: Memory Optimization via Coordinated Graph Transformation and Scheduling for DNN (ASPLOS'24)☆43Updated 5 months ago
- Implementation of TSM2L and TSM2R -- High-Performance Tall-and-Skinny Matrix-Matrix Multiplication Algorithms for CUDA☆31Updated 4 years ago
- ☆23Updated 10 months ago
- ☆33Updated 5 months ago
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆85Updated 8 months ago
- ☆41Updated 6 months ago