pytorch-labs / triton-cpuLinks
An experimental CPU backend for Triton (https//github.com/openai/triton)
☆42Updated 2 months ago
Alternatives and similar repositories for triton-cpu
Users that are interested in triton-cpu are comparing it to the libraries listed below
Sorting:
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆88Updated this week
- MLIR-based partitioning system☆86Updated this week
- A lightweight, Pythonic, frontend for MLIR☆81Updated last year
- Unified compiler/runtime for interfacing with PyTorch Dynamo.☆100Updated 2 weeks ago
- Ahead of Time (AOT) Triton Math Library☆64Updated this week
- An experimental CPU backend for Triton☆118Updated last month
- ☆79Updated 6 months ago
- An extension library of WMMA API (Tensor Core API)☆97Updated 10 months ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆150Updated this week
- Intel® Extension for MLIR. A staging ground for MLIR dialects and tools for Intel devices using the MLIR toolchain.☆135Updated this week
- ☆50Updated last year
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆109Updated 10 months ago
- ☆146Updated this week
- ☆96Updated last year
- Test suite for probing the numerical behavior of NVIDIA tensor cores☆38Updated 10 months ago
- Reference Kernels for the Leaderboard☆49Updated last week
- ☆36Updated this week
- extensible collectives library in triton☆87Updated 2 months ago
- A CUTLASS implementation using SYCL☆23Updated this week
- The missing pieces (as far as boilerplate reduction goes) of the upstream MLIR python bindings.☆99Updated last week
- High-Performance SGEMM on CUDA devices☆92Updated 4 months ago
- ☆27Updated last year
- Explore training for quantized models☆18Updated 4 months ago
- TPP experimentation on MLIR for linear algebra☆131Updated this week
- Experiments and prototypes associated with IREE or MLIR☆50Updated 9 months ago
- ☆34Updated last week
- IREE's PyTorch Frontend, based on Torch Dynamo.☆85Updated this week
- ☆44Updated 4 years ago
- ☆13Updated 2 months ago
- MatMul Performance Benchmarks for a Single CPU Core comparing both hand engineered and codegen kernels.☆130Updated last year