sgl-project / DeepGEMM
DeepGEMM: clean and efficient FP8 GEMM kernels with fine-grained scaling
☆15Updated last week
Alternatives and similar repositories for DeepGEMM:
Users that are interested in DeepGEMM are comparing it to the libraries listed below
- Quantized Attention on GPU☆45Updated 5 months ago
- ☆54Updated last week
- Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.☆36Updated 2 weeks ago
- An auxiliary project analysis of the characteristics of KV in DiT Attention.☆29Updated 4 months ago
- Multiple GEMM operators are constructed with cutlass to support LLM inference.☆17Updated 6 months ago
- [WIP] Better (FP8) attention for Hopper☆27Updated last month
- ☆58Updated this week
- ☆68Updated 2 months ago
- Open deep learning compiler stack for cpu, gpu and specialized accelerators☆18Updated this week
- Benchmark tests supporting the TiledCUDA library.☆16Updated 5 months ago
- A GPU-optimized system for efficient long-context LLMs decoding with low-bit KV cache.☆33Updated 3 weeks ago
- DeeperGEMM: crazy optimized version☆66Updated 2 weeks ago
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆85Updated this week
- A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters☆44Updated 8 months ago
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆81Updated last week
- ☆28Updated 2 months ago
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆70Updated 3 weeks ago
- [ACL 2024] RelayAttention for Efficient Large Language Model Serving with Long System Prompts☆39Updated last year
- Odysseus: Playground of LLM Sequence Parallelism☆68Updated 10 months ago
- TensorRT LLM Benchmark Configuration☆13Updated 8 months ago
- Framework to reduce autotune overhead to zero for well known deployments.☆63Updated this week
- GPTQ inference TVM kernel☆38Updated 11 months ago
- ☆81Updated 3 weeks ago
- Beyond KV Caching: Shared Attention for Efficient LLMs☆18Updated 9 months ago
- A practical way of learning Swizzle☆17Updated 2 months ago
- FlexAttention w/ FlashAttention3 Support☆26Updated 6 months ago
- ☆19Updated last month
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆111Updated this week
- A CUDA kernel for NHWC GroupNorm for PyTorch☆18Updated 5 months ago
- APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…☆23Updated this week