mirror of
https://github.com/ROCm/ROCm.git
synced 2026-04-05 03:01:17 -04:00
5b36cb48ad
Ref issue: - #1408 Changes: - Add `.editorconfig` - Add `pre-commit-hooks`: ```yaml - repo: https://github.com/pre-commit/pre-commit-hooks rev: v4.4.0 hooks: - id: check-symlinks - id: destroyed-symlinks - id: trailing-whitespace - id: end-of-file-fixer - id: check-yaml - id: check-toml - id: check-ast - id: check-added-large-files - id: check-merge-conflict - id: check-executables-have-shebangs - id: check-shebang-scripts-are-executable - id: detect-private-key - id: debug-statements ``` - Add `flake8` to `pre-commit` config and add `.flake8` file - Use `pre-commit` for style tests in CI - Run `pre-commit` and fix existing violations: - fix trailing spaces - fix end-of-files - fix mod file mode with `chmod -x` - run `autopep8` on existing code - fix `flake8` violations
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
|