ybai62868 / triton-mlir

Development repository for the Triton language and compiler

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Triton logo

Wheels

Documentation
Documentation

Triton

This is the development repository of Triton, a language and compiler for writing highly efficient custom Deep-Learning primitives. The aim of Triton is to provide an open-source environment to write fast code at higher productivity than CUDA, but also with higher flexibility than other existing DSLs.

The foundations of this project are described in the following MAPL2019 publication: Triton: An Intermediate Language and Compiler for Tiled Neural Network Computations. Please consider citing this work if you use Triton!

The official documentation contains installation instructions and tutorials.

Install from source

git clone https://github.com/openai/triton.git
cd triton
mkdir build && cd build
cmake .. 
make -j8
mv .libtriton.so /path-to-triton-python-_C/



# New to the mlir-yang-complier

About

Development repository for the Triton language and compiler

License:MIT License


Languages

Language:C++ 42.9%Language:Python 39.2%Language:MLIR 16.2%Language:CMake 1.2%Language:C 0.4%Language:Shell 0.1%Language:LLVM 0.0%Language:HTML 0.0%