install triton

ROCm · Jun 5, 2024 · 820e5df · 820e5df
1 parent baabd90
commit 820e5df
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/.github/workflows/amd_tests.yml b/.github/workflows/amd_tests.yml
@@ -47,6 +47,14 @@ jobs:
     steps:
       - name: Checkout
         uses: actions/checkout@v4
+      - name: Install Triton
+        run: |
+          pip uninstall -y triton
+          pip install matplotlib pandas pytest
+          git clone https://github.com/triton-lang/triton
+          cd triton
+          pip install --verbose -e python
+          cd ..
       - name: Build
         run: |
           FLASH_ATTENTION_SKIP_CUDA_BUILD=TRUE python setup.py install