Reinstall flash_attention if it is modified in the CI.

pytorch-labs · Dec 6, 2024 · 2b3f822 · 2b3f822
1 parent de24cc4
commit 2b3f822
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/.github/workflows/_linux-test-h100.yml b/.github/workflows/_linux-test-h100.yml
@@ -22,11 +22,23 @@ jobs:
         uses: actions/checkout@v3
         with:
           submodules: recursive
+      - uses: dorny/paths-filter@v3
+        id: flash_attn_updated
+        with:
+          filters: |
+            src:
+              - 'submodules/flash-attention/**'
       - name: Tune Nvidia GPU
         run: |
           sudo nvidia-smi -pm 1
           sudo ldconfig
           nvidia-smi
+      - name: Reinstall flash-attn (optional)
+        if: steps.flash_attn_updated
+        run: |
+          . "${SETUP_SCRIPT}"
+          python install.py --fa2
+          python install.py --fa3
       - name: Install Tritonbench
         run: |
           # speedup install and skip compile by reusing the docker .so files

diff --git a/submodules/flash-attention b/submodules/flash-attention