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
Sorting:
- The Tensor Algebra SuperOptimizer for Deep Learning☆739Jan 26, 2023Updated 3 years ago
- DietCode Code Release☆65Jul 21, 2022Updated 3 years ago
- A language and compiler for irregular tensor programs.☆152Nov 29, 2024Updated last year
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆143Mar 31, 2023Updated 2 years ago
- [MLSys 2021] IOS: Inter-Operator Scheduler for CNN Acceleration☆199Apr 27, 2022Updated 3 years ago
- Automatic Mapping Generation, Verification, and Exploration for ISA-based Spatial Accelerators☆121Oct 26, 2022Updated 3 years ago
- ☆145Jan 30, 2025Updated last year
- A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.☆1,006Sep 19, 2024Updated last year
- TiledKernel is a code generation library based on macro kernels and memory hierarchy graph data structure.☆19May 12, 2024Updated last year
- Re-implementation of the TASO compiler using equality saturation☆138Jun 28, 2021Updated 4 years ago
- Slides from 2021-12-15 talk, "TVM Developer Bootcamp – Writing Hardware Backends"☆11Jan 20, 2022Updated 4 years ago
- ☆192Mar 28, 2023Updated 2 years ago
- Tzer: TVM Implementation of "Coverage-Guided Tensor Compiler Fuzzing with Joint IR-Pass Mutation (OOPSLA'22)“.☆71Mar 9, 2023Updated 2 years ago
- ☆42Sep 8, 2023Updated 2 years ago
- Repository for SysML19 Artifacts Evaluation☆53Feb 28, 2019Updated 7 years ago
- Automatic Schedule Exploration and Optimization Framework for Tensor Computations☆182Apr 25, 2022Updated 3 years ago
- A schedule language for large model training☆152Aug 21, 2025Updated 6 months ago
- A list of tutorials, paper, talks, and open-source projects for emerging compiler and architecture☆519Jan 15, 2025Updated last year
- ☆289Feb 4, 2026Updated last month
- An external memory allocator example for PyTorch.☆16Aug 10, 2025Updated 6 months ago
- BladeDISC is an end-to-end DynamIc Shape Compiler project for machine learning workloads.☆917Dec 30, 2024Updated last year
- System for automated integration of deep learning backends.☆47Aug 15, 2022Updated 3 years ago
- ☆95Nov 4, 2022Updated 3 years ago
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆50Jul 23, 2024Updated last year
- A home for the final text of all TVM RFCs.☆109Sep 24, 2024Updated last year
- ☆84Dec 2, 2022Updated 3 years ago
- A list of awesome compiler projects and papers for tensor computation and deep learning.☆2,733Oct 19, 2024Updated last year
- This is a demo how to write a high performance convolution run on apple silicon☆57Feb 8, 2022Updated 4 years ago
- 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
- Lightweight and Parallel Deep Learning Framework☆264Nov 26, 2022Updated 3 years ago
- ☆70Jun 16, 2021Updated 4 years ago
- play gemm with tvm☆92Jul 22, 2023Updated 2 years ago
- Dorylus: Affordable, Scalable, and Accurate GNN Training☆76May 31, 2021Updated 4 years ago
- An open-source efficient deep learning framework/compiler, written in python.☆737Sep 4, 2025Updated 6 months ago
- ☆423Feb 24, 2026Updated last week
- A source-to-source compiler for optimizing CUDA dynamic parallelism by aggregating launches☆15Jun 21, 2019Updated 6 years ago
- An Attention Superoptimizer☆22Jan 20, 2025Updated last year
- Training neural networks in TensorFlow 2.0 with 5x less memory☆137Feb 21, 2022Updated 4 years ago
- Bridging polyhedral analysis tools to the MLIR framework☆119Sep 9, 2023Updated 2 years ago