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

add Windows build

Browse Source
This commit is contained in:
Junya Morioka
2025-05-27 22:47:05 +09:00
parent cf873af1fd
commit 67741fc910
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/build.yml
+5 -5
View File
@@ -27,7 +27,7 @@ jobs:
| Flash-Attention | Python | PyTorch | CUDA | | Flash-Attention | Python | PyTorch | CUDA |
| --- | --- | --- | --- | | --- | --- | --- | --- |
| 2.6.3 | 3.11 | 2.6.0 | 12.6.3 | | 2.4.3, 2.5.9, 2.6.3, 2.7.4 | 3.10, 3.11, 3.12 | 2.3.1, 2.4.1, 2.5.1, 2.6.0, 2.7.0 | 11.8.0, 12.4.1 |
# ######################################################### # #########################################################
# Build wheels with GitHub hosted runner # Build wheels with GitHub hosted runner
@@ -373,11 +373,11 @@ jobs:
strategy: strategy:
fail-fast: false fail-fast: false
matrix: matrix:
flash-attn-version: ["2.6.3"] flash-attn-version: ["2.4.3", "2.5.9", "2.6.3", "2.7.4"]
python-version: ["3.11"] python-version: ["3.10", "3.11", "3.12"]
torch-version: ["2.6.0"] torch-version: ["2.3.1", "2.4.1", "2.5.1", "2.6.0", "2.7.0"]
# https://developer.nvidia.com/cuda-toolkit-archive # https://developer.nvidia.com/cuda-toolkit-archive
cuda-version: ["12.6.3"] cuda-version: ["11.8.0", 12.4.1"]
exclude: exclude:
# torch < 2.2 does not support Python 3.12 # torch < 2.2 does not support Python 3.12
- python-version: "3.12" - python-version: "3.12"