heheda12345 / MagPy
☆38Updated 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 …☆176Updated last month
- ☆84Updated 6 months ago
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆51Updated 7 months ago
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆59Updated last week
- ASPLOS'24: Optimal Kernel Orchestration for Tensor Programs with Korch☆32Updated 7 months ago
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing. By pro…☆67Updated this week
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆119Updated 2 years ago
- ☆19Updated 5 months ago
- Artifacts of EVT ASPLOS'24☆23Updated last year
- High performance Transformer implementation in C++.☆105Updated last month
- DietCode Code Release☆61Updated 2 years ago
- Canvas: End-to-End Kernel Architecture Search in Neural Networks☆26Updated 3 months ago
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆89Updated 2 weeks ago
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆78Updated 3 months ago
- FlexFlow Serve: Low-Latency, High-Performance LLM Serving☆25Updated this week
- ☆42Updated last month
- MAGIS: Memory Optimization via Coordinated Graph Transformation and Scheduling for DNN (ASPLOS'24)☆49Updated 9 months ago
- nnScaler: Compiling DNN models for Parallel Training☆101Updated 3 weeks ago
- Implement Flash Attention using Cute.☆71Updated 2 months ago
- play gemm with tvm☆89Updated last year
- ☆27Updated 7 months ago
- A Vectorized N:M Format for Unleashing the Power of Sparse Tensor Cores☆50Updated last year
- ☆79Updated this week
- Standalone Flash Attention v2 kernel without libtorch dependency☆105Updated 6 months ago
- ☆137Updated 7 months 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
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆100Updated 8 months ago
- Automatic Mapping Generation, Verification, and Exploration for ISA-based Spatial Accelerators☆107Updated 2 years ago
- ☆75Updated 2 years ago
- Chimera: Efficiently Training Large-Scale Neural Networks with Bidirectional Pipelines.☆60Updated last year