Skip to content

Commit 66ae92e

Browse files
authored
Inline install triton (#364)
1 parent 52b98ef commit 66ae92e

File tree

2 files changed

+14
-24
lines changed

2 files changed

+14
-24
lines changed

.github/scripts/install_triton.sh

Lines changed: 0 additions & 23 deletions
This file was deleted.

.github/workflows/test.yml

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,20 @@ jobs:
6363
run: |
6464
apt-get update
6565
apt-get install -y git
66-
./.github/scripts/install_triton.sh
66+
mkdir -p /tmp/$USER
67+
cd /tmp/$USER
68+
pip uninstall -y triton pytorch-triton || true
69+
rm -rf triton/ || true
70+
git clone https://github.com/triton-lang/triton.git
71+
cd triton/
72+
conda config --set channel_priority strict
73+
conda install -y -c conda-forge conda=25.3.1 conda-libmamba-solver
74+
conda config --set solver libmamba
75+
conda install -y -c conda-forge gcc_linux-64=13 gxx_linux-64=13 gcc=13 gxx=13
76+
pip install -r python/requirements.txt
77+
MAX_JOBS=$(nproc) TRITON_PARALLEL_LINK_JOBS=2 pip install .
78+
cd /tmp/$USER
79+
rm -rf triton/
6780
6881
- name: Install Requirements
6982
run: |

0 commit comments

Comments
 (0)