TileLang / tvm
Open deep learning compiler stack for cpu, gpu and specialized accelerators
☆15Updated this week
Related projects ⓘ
Alternatives and complementary repositories for tvm
- TensorRT LLM Benchmark Configuration☆11Updated 3 months ago
- GPTQ inference TVM kernel☆36Updated 6 months ago
- Decoding Attention is specially optimized for multi head attention (MHA) using CUDA core for the decoding stage of LLM inference.☆25Updated 2 weeks ago
- 使用 CUDA C++ 实现的 llama 模型推理框架☆24Updated 2 weeks ago
- Quantized Attention on GPU☆30Updated 2 weeks ago
- ☆16Updated this week
- ☆18Updated last month
- Standalone Flash Attention v2 kernel without libtorch dependency☆98Updated 2 months ago
- Odysseus: Playground of LLM Sequence Parallelism☆57Updated 5 months ago
- Efficient, Flexible and Portable Structured Generation☆53Updated this week
- ☆79Updated 2 months ago
- 使用 cutlass 仓库在 ada 架构上实现 fp8 的 flash attention☆52Updated 3 months ago
- ☆47Updated 2 months ago
- ☆52Updated 2 weeks ago
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆85Updated 8 months ago
- FlexAttention w/ FlashAttention3 Support☆27Updated last month
- ☆35Updated 2 weeks ago
- APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…☆20Updated last week
- llama INT4 cuda inference with AWQ☆48Updated 4 months ago
- TiledCUDA is a highly efficient kernel template library designed to elevate CUDA C’s level of abstraction for processing tiles.☆156Updated this week
- (WIP) Parallel inference for black-forest-labs' FLUX model.☆11Updated this week
- [ACL 2024] RelayAttention for Efficient Large Language Model Serving with Long System Prompts☆34Updated 8 months ago
- This is a demo how to write a high performance convolution run on apple silicon☆52Updated 2 years ago
- Compare different hardware platforms via the Roofline Model for LLM inference tasks.☆75Updated 8 months ago
- FP64 equivalent GEMM via Int8 Tensor Cores using the Ozaki scheme☆46Updated 2 months ago
- Fast and memory-efficient exact attention☆30Updated 3 weeks ago
- Summary of system papers/frameworks/codes/tools on training or serving large model☆56Updated 11 months ago
- 分层解耦的深度学习推理引擎☆60Updated 2 months ago