mirror of
https://github.com/ROCm/ROCm.git
synced 2026-02-21 03:00:39 -05:00
d416d63152
Previously the nightly run was failing to upload if there had been no commits since the previous night. Also moves time ahead 20 minutes to avoid hourly spike delays launching workflows.
56 lines
1.2 KiB
ReStructuredText
56 lines
1.2 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.7-3.11 and PyPy 3.8-3.9.
|
|
|
|
And the latest nightly release:
|
|
|
|
.. code-block:: bash
|
|
|
|
pip install -U --index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/Triton-Nightly/pypi/simple/ triton-nightly
|
|
|
|
|
|
-----------
|
|
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
|