20100a7254
This PR merges the `triton-mlir` branch, in which we have been quietly rewriting the Triton backend from scratch to increase maintainability, stability and ultimately performance. Changes to the runtime are minimal, and this new version aims to remain backward-compatible with the previous commit. The legacy backend is now officially deprecated, but can still be accessed via the `legacy-backend` tag. Co-authored-by: Keren Zhou <kerenzhou@openai.com> Co-authored-by: Yan Chunwei <yanchunwei@outlook.com> Co-authored-by: goostavz <109190422+goostavz@users.noreply.github.com> Co-authored-by: Shintaro Iwasaki <siwasaki@fb.com> Co-authored-by: Yan Da <dyanab@connect.ust.hk> Co-authored-by: Jun Yang <yangjunpro@gmail.com> Co-authored-by: Ian Bearman <ianb@microsoft.com> Co-authored-by: Jason Ansel <jansel@jansel.net> Co-authored-by: Qingyi Liu <qingyil@nvidia.com> Co-authored-by: ben-zhang-609 <110140741+ben-zhang-609@users.noreply.github.com> Co-authored-by: Chenggang Zhao <lyricz@yeah.net> Co-authored-by: ben-zhang-609 <benzh609@gmail.com> Co-authored-by: dongdongl <dongdongl@nvidia.com>
56 lines
1.1 KiB
ReStructuredText
56 lines
1.1 KiB
ReStructuredText
==============
|
|
Installation
|
|
==============
|
|
|
|
---------------------
|
|
Binary Distributions
|
|
---------------------
|
|
|
|
You can install the latest stable release of Triton from pip:
|
|
|
|
.. code-block:: bash
|
|
|
|
pip install triton
|
|
|
|
Binary wheels are available for CPython 3.6-3.9 and PyPy 3.6-3.7.
|
|
|
|
And the latest nightly release:
|
|
|
|
.. code-block:: bash
|
|
|
|
pip install -U --pre triton
|
|
|
|
|
|
--------------
|
|
From Source
|
|
--------------
|
|
|
|
+++++++++++++++
|
|
Python Package
|
|
+++++++++++++++
|
|
|
|
You can install the Python package from source by running the following commands:
|
|
|
|
.. code-block:: bash
|
|
|
|
git clone https://github.com/openai/triton.git;
|
|
cd triton/python;
|
|
pip install cmake; # build time dependency
|
|
pip install -e .
|
|
|
|
Note that, if llvm-11 is not present on your system, the setup.py script will download the official LLVM11 static libraries link against that.
|
|
|
|
You can then test your installation by running the unit tests:
|
|
|
|
.. code-block:: bash
|
|
|
|
pip install -e '.[tests]'
|
|
pytest -vs test/unit/
|
|
|
|
and the benchmarks
|
|
|
|
.. code-block:: bash
|
|
|
|
cd bench/
|
|
python -m run --with-plots --result-dir /tmp/triton-bench
|