Skip to content

Commit

Permalink
Fix the triton main branch install
Browse files Browse the repository at this point in the history
  • Loading branch information
xuzhao9 committed Nov 21, 2024
1 parent 587f386 commit a5c5020
Showing 1 changed file with 12 additions and 0 deletions.
12 changes: 12 additions & 0 deletions .ci/tritonbench/install-triton-main.sh
Original file line number Diff line number Diff line change
Expand Up @@ -28,5 +28,17 @@ conda activate "${CONDA_ENV}"
cd /workspace
git clone https://github.com/triton-lang/triton.git
cd /workspace/triton
# delete the original triton directory
TRITON_PKG_DIR=$(python -c "import triton; import os; print(os.path.dirname(triton.__file__))")
# make sure all pytorch triton has been uninstalled
pip uninstall -y triton
pip uninstall -y triton
pip uninstall -y triton
rm -rf "${TRITON_PKG_DIR}"

# install main triton
pip install ninja cmake wheel pybind11; # build-time dependencies
pip install -e python

# test main branch installation with importing experimental descriptor
python -c "import triton.tools.experimental_descriptor"

0 comments on commit a5c5020

Please sign in to comment.