Files
triton/README.md

34 lines
1.2 KiB
Markdown
Raw Normal View History

# Triton
2020-05-10 00:34:10 -04:00
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](http://www.eecs.harvard.edu/~htk/publication/2019-mapl-tillet-kung-cox.pdf). Please consider citing us if you use our work!
2020-02-10 03:47:49 -05:00
## Installation
2020-05-07 10:48:41 -04:00
You can install the latest release with pip as follows:
```
2020-11-07 22:35:23 -05:00
sudo apt-get install llvm-10-dev
2020-05-07 10:48:41 -04:00
pip install triton
```
or the latest development version with:
```
pip install -e "git+https://github.com/ptillet/triton.git#egg=triton&subdirectory=python"
```
2020-05-10 00:34:10 -04:00
for the C++ package:
```
git clone https://github.com/ptillet/triton.git;
cd triton;
2020-05-10 00:34:10 -04:00
mkdir build;
cd build;
cmake ../;
make -j8;
```
2020-02-10 03:47:49 -05:00
## Getting Started
You can find tutorials for Triton for [Python](https://github.com/ptillet/triton/tree/master/tutorials) and [C++](https://github.com/ptillet/triton/tree/master/python/tutorials).