ROCm / aotriton
Ahead of Time (AOT) Triton Math Library
☆62Updated last week
Alternatives and similar repositories for aotriton:
Users that are interested in aotriton are comparing it to the libraries listed below
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆106Updated 9 months ago
- ☆78Updated 5 months ago
- ☆50Updated last year
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆84Updated this week
- extensible collectives library in triton☆85Updated last month
- ☆104Updated last month
- ☆202Updated 9 months ago
- ☆70Updated 4 months ago
- ☆30Updated this week
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆120Updated this week
- Standalone Flash Attention v2 kernel without libtorch dependency☆108Updated 7 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆87Updated this week
- Development repository for the Triton language and compiler☆118Updated this week
- ☆96Updated last year
- An experimental CPU backend for Triton (https//github.com/openai/triton)☆40Updated last month
- Benchmark code for the "Online normalizer calculation for softmax" paper☆91Updated 6 years ago
- Shared Middle-Layer for Triton Compilation☆246Updated 2 weeks ago
- Experimental projects related to TensorRT☆99Updated this week
- Applied AI experiments and examples for PyTorch☆262Updated last week
- ☆93Updated 7 months ago
- An extension library of WMMA API (Tensor Core API)☆96Updated 9 months ago
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆74Updated last month
- Test suite for probing the numerical behavior of NVIDIA tensor cores☆38Updated 9 months ago
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆91Updated this week
- We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …☆181Updated 3 months ago
- Several optimization methods of half-precision general matrix vector multiplication (HGEMV) using CUDA core.☆61Updated 7 months ago
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆135Updated 2 years ago
- DeeperGEMM: crazy optimized version☆68Updated this week
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆248Updated 6 months ago
- Unified compiler/runtime for interfacing with PyTorch Dynamo.☆100Updated 2 months ago