From d6870f54b0ad487cdb8ea5af17efa316c3f2027f Mon Sep 17 00:00:00 2001
From: Junya Morioka <77187490+mjun0812@users.noreply.github.com>
Date: Sun, 17 Nov 2024 02:36:15 +0900
Subject: [PATCH] add flash-attn 2.7.0.post2, and CUDA 12.3.2

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

diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 3e47796..cd4b4e3 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -25,7 +25,7 @@ jobs:
           body: |
             | 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 |
+            | 1.0.9, 2.4.3, 2.5.6, 2.5.9, 2.6.3, 2.7.0.post2 | 3.10, 3.11, 3.12 | 2.0.1, 2.1.2, 2.2.2, 2.3.1, 2.4.1, 2.5.1 | 11.8.0, 12.1.1, 12.3.2, 12.4.1 |
 
   build_wheels:
     name: Build wheels and Upload
@@ -34,10 +34,10 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        flash-attn-version: ["1.0.9", "2.4.3", "2.5.6", "2.5.9", "2.6.3"]
+        flash-attn-version: ["1.0.9", "2.4.3", "2.5.6", "2.5.9", "2.6.3", "2.7.0.post2"]
         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"]
+        torch-version: ["2.0.1", "2.1.2", "2.2.2", "2.3.1", "2.4.1", "2.5.1"]
+        cuda-version: ["11.8.0", "12.1.1", "12.3.2", "12.4.1"]
         exclude:
           # torch < 2.2 does not support Python 3.12
           - python-version: "3.12"