PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections
☆125Jun 23, 2022Updated 3 years ago
Alternatives and similar repositories for PET
Users that are interested in PET are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- The Tensor Algebra SuperOptimizer for Deep Learning☆740Jan 26, 2023Updated 3 years ago
- A language and compiler for irregular tensor programs.☆152Nov 29, 2024Updated last year
- DietCode Code Release☆65Jul 21, 2022Updated 3 years ago
- [MLSys 2021] IOS: Inter-Operator Scheduler for CNN Acceleration☆199Apr 27, 2022Updated 3 years ago
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆144Mar 31, 2023Updated 2 years ago
- Automatic Mapping Generation, Verification, and Exploration for ISA-based Spatial Accelerators☆122Oct 26, 2022Updated 3 years ago
- Re-implementation of the TASO compiler using equality saturation☆138Jun 28, 2021Updated 4 years ago
- A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.☆1,003Sep 19, 2024Updated last year
- ☆145Jan 30, 2025Updated last year
- TiledKernel is a code generation library based on macro kernels and memory hierarchy graph data structure.☆19May 12, 2024Updated last year
- Slides from 2021-12-15 talk, "TVM Developer Bootcamp – Writing Hardware Backends"☆11Jan 20, 2022Updated 4 years ago
- Tzer: TVM Implementation of "Coverage-Guided Tensor Compiler Fuzzing with Joint IR-Pass Mutation (OOPSLA'22)“.☆71Mar 9, 2023Updated 3 years ago
- ☆192Mar 28, 2023Updated 2 years ago
- Repository for SysML19 Artifacts Evaluation☆53Feb 28, 2019Updated 7 years ago
- InfiniTensor is a high-performance inference engine tailored for GPUs and AI accelerators. Its design focuses on effective deployment and…☆313Mar 16, 2026Updated last week
- ☆42Sep 8, 2023Updated 2 years ago
- System for automated integration of deep learning backends.☆47Aug 15, 2022Updated 3 years ago
- A list of tutorials, paper, talks, and open-source projects for emerging compiler and architecture☆520Jan 15, 2025Updated last year
- A home for the final text of all TVM RFCs.☆109Sep 24, 2024Updated last year
- A schedule language for large model training☆152Aug 21, 2025Updated 7 months ago
- Automatic Schedule Exploration and Optimization Framework for Tensor Computations☆183Apr 25, 2022Updated 3 years ago
- ASPLOS'24: Optimal Kernel Orchestration for Tensor Programs with Korch☆39Mar 27, 2025Updated 11 months ago
- BladeDISC is an end-to-end DynamIc Shape Compiler project for machine learning workloads.☆921Dec 30, 2024Updated last year
- ☆97Nov 4, 2022Updated 3 years ago
- An external memory allocator example for PyTorch.☆16Aug 10, 2025Updated 7 months ago
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆51Jul 23, 2024Updated last year
- A list of awesome compiler projects and papers for tensor computation and deep learning.☆2,733Oct 19, 2024Updated last year
- Graphiler is a compiler stack built on top of DGL and TorchScript which compiles GNNs defined using user-defined functions (UDFs) into ef…☆59Oct 3, 2022Updated 3 years ago
- ☆84Dec 2, 2022Updated 3 years ago
- Benchmark PyTorch Custom Operators☆14Jul 6, 2023Updated 2 years ago
- play gemm with tvm☆92Jul 22, 2023Updated 2 years ago
- ☆69Jun 16, 2021Updated 4 years ago
- An open-source efficient deep learning framework/compiler, written in python.☆739Sep 4, 2025Updated 6 months ago
- ☆41Jun 5, 2024Updated last year
- Dorylus: Affordable, Scalable, and Accurate GNN Training☆76May 31, 2021Updated 4 years ago
- Benchmark scripts for TVM☆74Mar 15, 2022Updated 4 years ago
- Antares: an automatic engine for multi-platform kernel generation and optimization. Supporting CPU, CUDA, ROCm, DirectX12, GraphCore, SYC…☆465Apr 20, 2025Updated 11 months ago
- Lightweight and Parallel Deep Learning Framework☆264Nov 26, 2022Updated 3 years ago
- An experimental ahead of time compiler for Relay.☆49Apr 21, 2020Updated 5 years ago