Installation

Jump to bottom

Xu Zhao edited this page Oct 17, 2025 · 2 revisions

We manage TritonBench install with install.py.

Basic Usage

If you run python install.py without any argument, it will do the following:

If PyTorch is missing, install the latest PyTorch nightly package.
Install other dependencies from requirements.txt and requirements_numpy.txt.
Checkout all submodules

Advanced Usage

install.py supports arguments that enable users to install optional third-party kernel libraries.

--liger: Install liger-kernel-nightly package (Triton).
--fa3: Install Flash Attention 3 (CUTLASS).
--fa2: Install Flash Attention 2 (CUTLASS).
--fbgemm: Install FBGEMM GenAI kernels (CUTLASS and Triton).
--jax: Install JAX (Pallas and Mosaic).
--tk: Install ThunderKittens.
--tile: Install Tile Lang.
--xformers: Install XFormers.
--aiter: Install AITer.
--all: Install all of the above.