From 820e5dfb11946ffaa840845a740558a750adf7e9 Mon Sep 17 00:00:00 2001
From: Michael Melesse <micmelesse@gmail.com>
Date: Wed, 5 Jun 2024 11:39:09 -0500
Subject: [PATCH] install triton

---
 .github/workflows/amd_tests.yml | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/.github/workflows/amd_tests.yml b/.github/workflows/amd_tests.yml
index 3a498b079..7b7cde502 100644
--- 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