skip output test_flash_attn_output

ROCm · Aug 9, 2024 · 2e22858 · 2e22858
1 parent ebb3bd6
commit 2e22858
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/.github/workflows/amd_tests.yml b/.github/workflows/amd_tests.yml
@@ -57,13 +57,13 @@ jobs:
       - name: Build
         run: |
           python setup.py install
-      # - name: Flash Attention qkvpacked Tests
-      #   run: |
-      #     pytest tests/test_flash_attn.py::test_flash_attn_qkvpacked
-      #     pytest tests/test_flash_attn.py::test_flash_attn_varlen_qkvpacked
+      - name: Flash Attention qkvpacked Tests
+        run: |
+          pytest tests/test_flash_attn.py::test_flash_attn_qkvpacked
+          pytest tests/test_flash_attn.py::test_flash_attn_varlen_qkvpacked
       - name: Flash Attention output Tests
         run: |
-          pytest tests/test_flash_attn.py::test_flash_attn_output
+          # pytest tests/test_flash_attn.py::test_flash_attn_output
           pytest tests/test_flash_attn.py::test_flash_attn_varlen_output
       - name: Flash Attention causal Tests
         run: |