yindo/flash-attention-prebuild-wheels-rocm
1
0
Fork 0
mirror of https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git synced 2026-07-01 01:37:53 -04:00
Code Issues Packages Projects Releases 2 Wiki Activity

Update build.yml

Browse Source
This commit is contained in:
Junya Morioka
2025-02-16 03:46:47 +09:00
committed by GitHub
parent fae44bf9c5
commit 94701e68ba
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/build.yml
+3 -3
View File
@@ -25,7 +25,7 @@ jobs:
body: |
| Flash-Attention | Python | PyTorch | CUDA |
|-----------------|--------|---------|------|
| 2.6.3, 2.7.4.post1 | 3.10, 3.11, 3.12 | 2.0.1, 2.1.2, 2.2.2, 2.3.1, 2.4.1, 2.5.1, 2.6.0 | 12.4.1, 12.6.3 |
| "2.4.3", 2.5.9, 2.6.3, 2.7.4.post1 | 3.10, 3.11, 3.12 | 2.2.2, 2.3.1, 2.4.1, 2.5.1, 2.6.0 | 12.4.1, 12.6.3 |
build_wheels:
name: Build wheels and Upload
@@ -34,10 +34,10 @@ jobs:
strategy:
fail-fast: false
matrix:
flash-attn-version: ["2.6.3", "2.7.4.post1"]
flash-attn-version: ["2.4.3", "2.5.9", "2.6.3", "2.7.4.post1"]
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.1", "2.6.0"]
["2.2.2", "2.3.1", "2.4.1", "2.5.1", "2.6.0"]
cuda-version: ["12.4.1", "12.6.3"]
exclude:
# torch < 2.2 does not support Python 3.12