add Python 3.10, flash-attn v1.0.9

mjun0812 · Nov 1, 2024 · c8497c1 · c8497c1
1 parent 06cd096
commit c8497c1
Showing 1 changed file with 5 additions and 21 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -23,25 +23,9 @@ jobs:
           tag_name: ${{ steps.extract_branch.outputs.branch }}
           release_name: ${{ steps.extract_branch.outputs.branch }}
           body: |
-            - Flash-Attention
-              - 2.4.3
-              - 2.5.6
-              - 2.5.9
-              - 2.6.3
-            - Python
-              - 3.11
-              - 3.12
-            - PyTorch
-              - 2.0.1
-              - 2.1.2
-              - 2.2.2
-              - 2.3.1
-              - 2.4.1
-              - 2.5.0
-            - CUDA
-              - 11.8.0
-              - 12.1.1
-              - 12.4.1
+            | Flash-Attention | Python | PyTorch | CUDA |
+            |-----------------|--------|---------|------|
+            | 1.0.9, 2.4.3, 2.5.6, 2.5.9, 2.6.3 | 3.10, 3.11, 3.12 | 2.0.1, 2.1.2, 2.2.2, 2.3.1, 2.4.1, 2.5.0 | 11.8.0, 12.1.1, 12.4.1 |
 
   build_wheels:
     name: Build wheels and Upload
@@ -50,8 +34,8 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        flash-attn-version: ["2.4.3", "2.5.6", "2.5.9", "2.6.3"]
-        python-version: ["3.11", "3.12"]
+        flash-attn-version: ["1.0.9", "2.4.3", "2.5.6", "2.5.9", "2.6.3"]
+        python-version: ["3.10", "3.11", "3.12"]
         torch-version: ["2.0.1", "2.1.2", "2.2.2", "2.3.1", "2.4.1", "2.5.0"]
         cuda-version: ["11.8.0", "12.1.1", "12.4.1"]
         exclude: