pytorch-labs / triton-cpu
An experimental CPU backend for Triton (https//github.com/openai/triton)
☆40Updated last month
Alternatives and similar repositories for triton-cpu:
Users that are interested in triton-cpu are comparing it to the libraries listed below
- MLIR-based partitioning system☆82Updated this week
- 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
- Unified compiler/runtime for interfacing with PyTorch Dynamo.☆100Updated 2 months ago
- ☆78Updated 6 months ago
- ☆50Updated last year
- Ahead of Time (AOT) Triton Math Library☆62Updated last week
- IREE's PyTorch Frontend, based on Torch Dynamo.☆82Updated this week
- An experimental CPU backend for Triton☆109Updated last week
- High-Performance SGEMM on CUDA devices☆90Updated 3 months ago
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆106Updated 9 months ago
- ☆30Updated this week
- An extension library of WMMA API (Tensor Core API)☆96Updated 9 months ago
- Experiments and prototypes associated with IREE or MLIR☆50Updated 8 months ago
- ☆96Updated last year
- A lightweight, Pythonic, frontend for MLIR☆81Updated last year
- ☆104Updated last month
- The missing pieces (as far as boilerplate reduction goes) of the upstream MLIR python bindings.☆90Updated this week
- ☆142Updated this week
- ☆32Updated last week
- Reference Kernels for the Leaderboard☆42Updated this week
- Framework to reduce autotune overhead to zero for well known deployments.☆70Updated 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
- Shared Middle-Layer for Triton Compilation☆246Updated 2 weeks ago
- FractalTensor is a programming framework that introduces a novel approach to organizing data in deep neural networks (DNNs) as a list of …☆26Updated 4 months ago
- ☆13Updated last month
- Intel® Extension for MLIR. A staging ground for MLIR dialects and tools for Intel devices using the MLIR toolchain.☆134Updated last week
- Standalone Flash Attention v2 kernel without libtorch dependency☆108Updated 7 months ago
- Development repository for the Triton language and compiler☆118Updated this week
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆122Updated this week