thunlp / TritonBench
TritonBench: Benchmarking Large Language Model Capabilities for Generating Triton Operators
☆32Updated 3 weeks ago
Alternatives and similar repositories for TritonBench:
Users that are interested in TritonBench are comparing it to the libraries listed below
- ☆73Updated 4 months ago
- Implement Flash Attention using Cute.☆74Updated 3 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆77Updated 5 months ago
- GPTQ inference TVM kernel☆38Updated 11 months ago
- Estimate MFU for DeepSeekV3☆21Updated 2 months ago
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆74Updated this week
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆104Updated this week
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆102Updated 8 months ago
- A GPU-optimized system for efficient long-context LLMs decoding with low-bit KV cache.☆23Updated this week
- Boosting 4-bit inference kernels with 2:4 Sparsity☆71Updated 6 months ago
- 16-fold memory access reduction with nearly no loss☆86Updated this week
- We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …☆179Updated 2 months ago
- [ICLR 2025] TidalDecode: A Fast and Accurate LLM Decoding with Position Persistent Sparse Attention☆31Updated 3 weeks ago
- ☆67Updated 2 months ago
- A minimal implementation of vllm.☆37Updated 8 months ago
- DeeperGEMM: crazy optimized version☆63Updated 2 weeks ago
- Quantized Attention on GPU☆45Updated 4 months ago
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆63Updated last week
- ☆137Updated 8 months ago
- Odysseus: Playground of LLM Sequence Parallelism☆68Updated 9 months ago
- extensible collectives library in triton☆84Updated 6 months ago
- ☆102Updated 7 months ago
- ☆52Updated last year
- ☆52Updated this week
- ☆90Updated 6 months ago
- ☆61Updated 4 months ago
- ArkVale: Efficient Generative LLM Inference with Recallable Key-Value Eviction (NIPS'24)☆32Updated 3 months ago
- Standalone Flash Attention v2 kernel without libtorch dependency☆108Updated 6 months ago
- [NeurIPS 2024] Efficient LLM Scheduling by Learning to Rank☆43Updated 4 months ago
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆78Updated 4 months ago