tile-ai / tvm
Open deep learning compiler stack for cpu, gpu and specialized accelerators
☆17Updated 2 weeks ago
Alternatives and similar repositories for tvm:
Users that are interested in tvm are comparing it to the libraries listed below
- ☆39Updated this week
- TensorRT LLM Benchmark Configuration☆12Updated 5 months ago
- GPTQ inference TVM kernel☆38Updated 8 months ago
- Decoding Attention is specially optimized for multi head attention (MHA) using CUDA core for the decoding stage of LLM inference.☆27Updated 2 months ago
- Quantized Attention on GPU☆34Updated last month
- 📚[WIP] FFPA: Yet antother Faster Flash Prefill Attention with O(1)⚡️GPU SRAM complexity for headdim > 256, 1.8x~3x↑🎉faster vs SDPA EA.☆49Updated this week
- Framework to reduce autotune overhead to zero for well known deployments.☆57Updated 2 months ago
- 使用 CUDA C++ 实现的 llama 模型推理框架☆43Updated 2 months ago
- ☆24Updated this week
- ☆21Updated this week
- ☆22Updated 3 weeks ago
- Implement Flash Attention using Cute.☆65Updated last month
- Multiple GEMM operators are constructed with cutlass to support LLM inference.☆16Updated 3 months ago
- ☆19Updated 3 months ago
- ⚡ ️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, achieve peak⚡️ performance☆43Updated last week
- Standalone Flash Attention v2 kernel without libtorch dependency☆99Updated 4 months ago
- ☆59Updated 2 months ago
- Benchmark tests supporting the TiledCUDA library.☆12Updated 2 months ago
- ☆56Updated 3 months ago
- APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…☆21Updated 3 weeks ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆75Updated this week
- ☆78Updated 4 months ago
- An auxiliary project analysis of the characteristics of KV in DiT Attention.☆23Updated last month
- Odysseus: Playground of LLM Sequence Parallelism☆64Updated 7 months ago
- ☆57Updated 7 months ago
- A CUDA kernel for NHWC GroupNorm for PyTorch☆16Updated 2 months ago
- 使用 cutlass 仓库在 ada 架构上实现 fp8 的 flash attention☆52Updated 5 months ago
- Implementation of IceFormer: Accelerated Inference with Long-Sequence Transformers on CPUs (ICLR 2024).☆22Updated 7 months ago
- FlexAttention w/ FlashAttention3 Support☆27Updated 3 months ago