Install PyTorch's triton.

pytorch · Feb 5, 2025 · 75f1deb · 75f1deb
1 parent eba99a5
commit 75f1deb
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/.github/workflows/_test_requiring_torch_cuda.yml b/.github/workflows/_test_requiring_torch_cuda.yml
@@ -94,8 +94,12 @@ jobs:
           pip install -U --pre jaxlib -f https://storage.googleapis.com/jax-releases/jaxlib_nightly_releases.html
           pip install -U --pre jax-cuda12-pjrt jax-cuda12-plugin -f https://storage.googleapis.com/jax-releases/jax_cuda_plugin_nightly_releases.html
           pip install -U --pre jax -f https://storage.googleapis.com/jax-releases/jax_nightly_releases.html
-          pip install --no-deps triton==2.3.0
         if: ${{ matrix.run_triton_tests }}
+      - name: Install Triton
+        shell: bash
+        run: |
+          cd pytorch
+          make triton
       - name: Python Tests
         shell: bash
         run: |