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

fix: update PyTorch and CUDA version compatibility rules

Browse Source 
- Refine exclusion logic in create_matrix.py to match actual support
- Update CUDA version support for PyTorch 2.9 and 2.10
- Remove temporary build exclusions
This commit is contained in:
Junya Morioka
2026-01-27 18:33:34 +09:00
parent d32a4c9b60
commit 6e93ad0663
2 changed files with 27 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
create_matrix.py
+25 -22
View File
@@ -4,39 +4,42 @@ EXCLUDE = [
# 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.1.2"},
# TODO: Temporary exclusion - already built
{"python-version": "3.12", "torch-version": "2.9.1", "cuda-version": "12.8"},
{"python-version": "3.10", "torch-version": "2.7.1", "cuda-version": "12.8"},
{"python-version": "3.13", "torch-version": "2.7.1", "cuda-version": "12.8"},
{"python-version": "3.10", "torch-version": "2.8.0", "cuda-version": "12.8"},
{"python-version": "3.11", "torch-version": "2.5.1", "cuda-version": "12.8"},
# 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.4"},
{"torch-version": "2.0.1", "cuda-version": "12.6"},
{"torch-version": "2.0.1", "cuda-version": "12.8"},
# torch 2.6.0 does not support CUDA 12.1
{"torch-version": "2.6.0", "cuda-version": "12.1"},
# torch 2.7.0 does not support CUDA 12.4
{"torch-version": "2.7.0", "cuda-version": "12.4"},
# torch < 2.8 does not support CUDA 12.9
# torch 2.5.1 only supports CUDA 11.8, 12.1, 12.4
{"torch-version": "2.5.1", "cuda-version": "12.6"},
{"torch-version": "2.5.1", "cuda-version": "12.8"},
{"torch-version": "2.5.1", "cuda-version": "12.9"},
{"torch-version": "2.6.3", "cuda-version": "12.9"},
{"torch-version": "2.5.1", "cuda-version": "13.0"},
# torch 2.6.0 only supports CUDA 11.8, 12.4, 12.6
{"torch-version": "2.6.0", "cuda-version": "12.1"},
{"torch-version": "2.6.0", "cuda-version": "12.8"},
{"torch-version": "2.6.0", "cuda-version": "12.9"},
{"torch-version": "2.6.0", "cuda-version": "13.0"},
# torch 2.7.1 only supports CUDA 11.8, 12.6, 12.8
{"torch-version": "2.7.1", "cuda-version": "12.4"},
{"torch-version": "2.7.1", "cuda-version": "12.9"},
{"torch-version": "2.7.1", "cuda-version": "13.0"},
# torch 2.8.0 only supports CUDA 12.6, 12.8, 12.9
{"torch-version": "2.8.0", "cuda-version": "12.4"},
{"torch-version": "2.8.0", "cuda-version": "13.0"},
# torch 2.9.1 only supports CUDA 12.6, 12.8, 13.0
{"torch-version": "2.9.1", "cuda-version": "12.4"},
{"torch-version": "2.9.1", "cuda-version": "12.9"},
# torch 2.10.0 only supports CUDA 12.6, 12.8, 13.0
{"torch-version": "2.10.0", "cuda-version": "12.4"},
{"torch-version": "2.10.0", "cuda-version": "12.9"},
# torch < 2.6 does not support Python 3.13
{"torch-version": "2.5.1", "python-version": "3.13"},
# torch >= 2.9 does not support Python 3.9
{"torch-version": "2.9.1", "python-version": "3.9"},
{"torch-version": "2.10.0", "python-version": "3.9"},
# torch < 2.9 does not support CUDA 13.0
{"torch-version": "2.5.1", "cuda-version": "13.0"},
{"torch-version": "2.6.0", "cuda-version": "13.0"},
{"torch-version": "2.7.1", "cuda-version": "13.0"},
{"torch-version": "2.8.1", "cuda-version": "13.0"},
{"torch-version": "2.8.0", "cuda-version": "13.0"},
# torch 2.10 does not support CUDA 12.4
{"torch-version": "2.10.0", "cuda-version": "12.4"},
# Python 3.14 is supported from torch 2.9
# torch < 2.9 does not support Python 3.14
{"torch-version": "2.5.1", "python-version": "3.14"},
{"torch-version": "2.6.3", "python-version": "3.14"},
{"torch-version": "2.6.0", "python-version": "3.14"},
{"torch-version": "2.7.1", "python-version": "3.14"},
{"torch-version": "2.8.0", "python-version": "3.14"},
]