heheda12345 / MagPy
☆39Updated 9 months ago
Alternatives and similar repositories for MagPy:
Users that are interested in MagPy are comparing it to the libraries listed below
- We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …☆179Updated last month
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆62Updated 3 weeks ago
- Artifacts of EVT ASPLOS'24☆24Updated last year
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆51Updated 8 months ago
- ASPLOS'24: Optimal Kernel Orchestration for Tensor Programs with Korch☆32Updated 7 months ago
- ☆88Updated 6 months ago
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆119Updated 2 years ago
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆78Updated 4 months ago
- DietCode Code Release☆62Updated 2 years ago
- Implement Flash Attention using Cute.☆74Updated 3 months ago
- ☆21Updated last week
- ☆90Updated 2 weeks ago
- A GPU-optimized system for efficient long-context LLMs decoding with low-bit KV cache.☆23Updated this week
- ☆27Updated 8 months ago
- Canvas: End-to-End Kernel Architecture Search in Neural Networks☆26Updated 4 months ago
- ☆48Updated 2 months ago
- Triton to TVM transpiler.☆19Updated 5 months ago
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing. By pro…☆70Updated this week
- Magicube is a high-performance library for quantized sparse matrix operations (SpMM and SDDMM) of deep learning on Tensor Cores.☆86Updated 2 years ago
- play gemm with tvm☆89Updated last year
- Examples of CUDA implementations by Cutlass CuTe☆146Updated last month
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆135Updated last year
- MAGIS: Memory Optimization via Coordinated Graph Transformation and Scheduling for DNN (ASPLOS'24)☆49Updated 9 months ago
- ☆19Updated 5 months ago
- DeepSeek-V3/R1 inference performance simulator☆76Updated this week
- 使用 cutlass 仓库在 ada 架构上实现 fp8 的 flash attention☆60Updated 7 months ago
- Standalone Flash Attention v2 kernel without libtorch dependency☆106Updated 6 months ago
- Automatic Mapping Generation, Verification, and Exploration for ISA-based Spatial Accelerators☆108Updated 2 years ago
- ☆76Updated 2 years ago
- High performance Transformer implementation in C++.☆111Updated 2 months ago