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

ci: add Windows CUDA 12.8 builds for all Python and PyTorch versions

Browse Source 
Add the following combinations to windows_self_hosted matrix:
- Python: 3.10, 3.11, 3.12, 3.13
- PyTorch: 2.5.1, 2.6.0, 2.7.1, 2.8.0, 2.9.1
- CUDA: 12.8

Temporarily exclude 3.12 + 2.9.1 + 12.8 (already built in v0.7.6).
This commit is contained in:
Junya Morioka
2026-01-09 03:17:19 +09:00
parent 353c319d63
commit 567413ae32
1 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
create_matrix.py
+9 -7
View File
@@ -4,6 +4,8 @@ EXCLUDE = [
# torch < 2.2 does not support Python 3.12 # torch < 2.2 does not support Python 3.12
{"python-version": "3.12", "torch-version": "2.0.1"}, {"python-version": "3.12", "torch-version": "2.0.1"},
{"python-version": "3.12", "torch-version": "2.1.2"}, {"python-version": "3.12", "torch-version": "2.1.2"},
# TODO: Temporary exclusion - already built in v0.7.6
{"python-version": "3.12", "torch-version": "2.9.1", "cuda-version": "12.8"},
# torch 2.0.1 does not support CUDA 12.x # torch 2.0.1 does not support CUDA 12.x
{"torch-version": "2.0.1", "cuda-version": "12.1"}, {"torch-version": "2.0.1", "cuda-version": "12.1"},
{"torch-version": "2.0.1", "cuda-version": "12.4"}, {"torch-version": "2.0.1", "cuda-version": "12.4"},
@@ -202,16 +204,16 @@ WINDOWS_SELF_HOSTED_MATRIX = {
"2.8.3", "2.8.3",
], ],
"python-version": [ "python-version": [
# "3.10", "3.10",
# "3.11", "3.11",
"3.12", "3.12",
# "3.13", "3.13",
], ],
"torch-version": [ "torch-version": [
# "2.5.1", "2.5.1",
# "2.6.0", "2.6.0",
# "2.7.1", "2.7.1",
# "2.8.0", "2.8.0",
"2.9.1", "2.9.1",
], ],
"cuda-version": [ "cuda-version": [