DefTruth / hgemm-tensorcores-mma
⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.
☆52Updated 2 weeks ago
Alternatives and similar repositories for hgemm-tensorcores-mma:
Users that are interested in hgemm-tensorcores-mma are comparing it to the libraries listed below
- Implement Flash Attention using Cute.☆69Updated 2 months ago
- 使用 cutlass 仓库在 ada 架构上实现 fp8 的 flash attention☆53Updated 6 months ago
- ☆36Updated last month
- We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …☆175Updated 3 weeks ago
- Standalone Flash Attention v2 kernel without libtorch dependency☆104Updated 5 months ago
- Quantized Attention on GPU☆34Updated 2 months ago
- ☆81Updated 5 months ago
- TileFusion is a highly efficient kernel template library designed to elevate the level of abstraction in CUDA C for processing tiles.☆56Updated this week
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆88Updated 11 months ago
- Several optimization methods of half-precision general matrix vector multiplication (HGEMV) using CUDA core.☆55Updated 5 months ago
- ☆44Updated last month
- Examples of CUDA implementations by Cutlass CuTe☆138Updated 2 weeks ago
- ☆72Updated 2 months ago
- High performance Transformer implementation in C++.☆102Updated last month
- Decoding Attention is specially optimized for multi head attention (MHA) using CUDA core for the decoding stage of LLM inference.☆29Updated 3 months ago
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆78Updated 3 months ago
- Multiple GEMM operators are constructed with cutlass to support LLM inference.☆16Updated 4 months ago
- ☆19Updated 4 months ago
- ☆98Updated 2 months ago
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆35Updated 5 months ago
- ☆26Updated 10 months ago
- Tutorials of Extending and importing TVM with CMAKE Include dependency.☆13Updated 4 months ago
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆97Updated 7 months ago
- 📚FFPA: Yet another Faster Flash Prefill Attention with O(1)⚡️SRAM complexity for headdim > 256, 1.8x~3x↑🎉faster than SDPA EA.☆106Updated this week
- ☆38Updated 8 months ago
- ☆181Updated 7 months ago
- play gemm with tvm☆87Updated last year
- ☆61Updated 3 weeks ago
- ☆80Updated last year
- PyTorch bindings for CUTLASS grouped GEMM.☆64Updated 3 months ago