try upstream triton

ROCm · Jan 22, 2025 · 15cb129 · 15cb129
1 parent e50bc0c
commit 15cb129
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/.github/workflows/amd_tests.yml b/.github/workflows/amd_tests.yml
@@ -35,12 +35,14 @@ jobs:
       - name: Install Triton
         run: |
           pip uninstall -y triton
-          pip install matplotlib pandas pytest
           git clone https://github.com/triton-lang/triton
           cd triton
-          git checkout 3ca2f498e98ed7249b82722587c511a5610e00c4 
+          pip install ninja cmake wheel pybind11 # build-time dependencies
           pip install --verbose -e python
           cd ..
+      - name: Show Triton version
+        run: |
+          pip show triton
       - name: Build
         run: |
           export FLASH_ATTENTION_TRITON_AMD_ENABLE="TRUE"