pytorch-labs / triton-cpu
An experimental CPU backend for Triton (https//github.com/openai/triton)
☆39Updated 10 months ago
Alternatives and similar repositories for triton-cpu:
Users that are interested in triton-cpu are comparing it to the libraries listed below
- An experimental CPU backend for Triton☆99Updated this week
- Unified compiler/runtime for interfacing with PyTorch Dynamo.☆99Updated 2 weeks ago
- extensible collectives library in triton☆83Updated 5 months ago
- ☆73Updated 4 months ago
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing. By pro…☆67Updated this week
- MLIR-based partitioning system☆71Updated this week
- A lightweight, Pythonic, frontend for MLIR☆80Updated last year
- ☆49Updated last year
- An extension library of WMMA API (Tensor Core API)☆91Updated 8 months ago
- ☆79Updated this week
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆100Updated 8 months ago
- Experiments and prototypes associated with IREE or MLIR☆50Updated 7 months ago
- Ahead of Time (AOT) Triton Math Library☆54Updated this week
- Test suite for probing the numerical behavior of NVIDIA tensor cores☆37Updated 7 months ago
- High-Performance SGEMM on CUDA devices☆86Updated last month
- The missing pieces (as far as boilerplate reduction goes) of the upstream MLIR python bindings.☆83Updated this week
- ☆90Updated 10 months ago
- OpenAI Triton backend for Intel® GPUs☆168Updated this week
- Shared Middle-Layer for Triton Compilation☆230Updated this week
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆103Updated this week
- IREE's PyTorch Frontend, based on Torch Dynamo.☆71Updated this week
- TPP experimentation on MLIR for linear algebra☆120Updated this week
- Intel® Extension for MLIR. A staging ground for MLIR dialects and tools for Intel devices using the MLIR toolchain.☆130Updated this week
- Explore training for quantized models☆16Updated 2 months ago
- ☆22Updated this week
- ☆53Updated 2 months ago
- ☆25Updated last week
- We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …☆176Updated last month
- ☆137Updated this week
- ☆187Updated 8 months ago