TiledTensor / TiledCUDA
TiledCUDA is a highly efficient kernel template library designed to elevate CUDA C’s level of abstraction for processing tiles.
☆114Updated last week
Related projects: ⓘ
- play gemm with tvm☆81Updated last year
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆82Updated 6 months ago
- Standalone Flash Attention v2 kernel without libtorch dependency☆93Updated last week
- ☆138Updated 2 months ago
- ☆151Updated this week
- 使用 cutlass 仓库在 ada 架构上实现 fp8 的 flash attention☆46Updated last month
- ☆67Updated last week
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆81Updated 2 months ago
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆49Updated last month
- Several optimization methods of half-precision general matrix vector multiplication (HGEMV) using CUDA core.☆40Updated last week
- ☆70Updated 6 months ago
- A Easy-to-understand TensorOp Matmul Tutorial☆265Updated this week
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆20Updated last week
- ☆77Updated last year
- PyTorch bindings for CUTLASS grouped GEMM.☆41Updated 3 weeks ago
- A language and compiler for irregular tensor programs.☆132Updated 4 months ago
- CUDA Matrix Multiplication Optimization☆118Updated 2 months ago
- A fast communication-overlapping library for tensor parallelism on GPUs.☆184Updated this week
- flash attention tutorial written in python, triton, cuda, cutlass☆159Updated 3 months ago
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆129Updated last year
- High performance Transformer implementation in C++.☆67Updated this week
- GPTQ inference TVM kernel☆35Updated 4 months ago
- ☆134Updated last year
- An extension library of WMMA API (Tensor Core API)☆81Updated 2 months ago
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆112Updated 2 years ago
- High Performance Grouped GEMM in PyTorch☆20Updated 2 years ago
- llama INT4 cuda inference with AWQ☆46Updated 2 months ago
- Shared Middle-Layer for Triton Compilation☆160Updated this week
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆70Updated last month
- This is a demo how to write a high performance convolution run on apple silicon☆52Updated 2 years ago