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

feat: Update arm matrix

Browse Source
This commit is contained in:
Junya Morioka
2025-12-05 14:53:34 +09:00
parent 70f60421a2
commit e6542f9d80
1 changed files with 63 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files
create_matrix.py
+63 -37
View File
@@ -5,26 +5,26 @@ EXCLUDE = [
{"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"},
# 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.1"}, {"torch-version": "2.0.1", "cuda-version": "12.1"},
{"torch-version": "2.0.1", "cuda-version": "12.4.1"}, {"torch-version": "2.0.1", "cuda-version": "12.4"},
{"torch-version": "2.0.1", "cuda-version": "12.6.3"}, {"torch-version": "2.0.1", "cuda-version": "12.6"},
{"torch-version": "2.0.1", "cuda-version": "12.8.1"}, {"torch-version": "2.0.1", "cuda-version": "12.8"},
# torch 2.6.0 does not support CUDA 12.1 # torch 2.6.0 does not support CUDA 12.1
{"torch-version": "2.6.0", "cuda-version": "12.1.1"}, {"torch-version": "2.6.0", "cuda-version": "12.1"},
# torch 2.7.0 does not support CUDA 12.4 # torch 2.7.0 does not support CUDA 12.4
{"torch-version": "2.7.0", "cuda-version": "12.4.1"}, {"torch-version": "2.7.0", "cuda-version": "12.4"},
# torch < 2.8 does not support CUDA 12.9 # torch < 2.8 does not support CUDA 12.9
{"torch-version": "2.5.1", "cuda-version": "12.9.1"}, {"torch-version": "2.5.1", "cuda-version": "12.9"},
{"torch-version": "2.6.3", "cuda-version": "12.9.1"}, {"torch-version": "2.6.3", "cuda-version": "12.9"},
{"torch-version": "2.7.1", "cuda-version": "12.9.1"}, {"torch-version": "2.7.1", "cuda-version": "12.9"},
# torch >= 2.9 does not support Python 3.9 # torch >= 2.9 does not support Python 3.9
{"torch-version": "2.9.1", "python-version": "3.9"}, {"torch-version": "2.9.1", "python-version": "3.9"},
# torch < 2.9 does not support CUDA 13.0 # torch < 2.9 does not support CUDA 13.0
{"torch-version": "2.5.1", "cuda-version": "13.0.1"}, {"torch-version": "2.5.1", "cuda-version": "13.0"},
{"torch-version": "2.6.0", "cuda-version": "13.0.1"}, {"torch-version": "2.6.0", "cuda-version": "13.0"},
{"torch-version": "2.7.1", "cuda-version": "13.0.1"}, {"torch-version": "2.7.1", "cuda-version": "13.0"},
{"torch-version": "2.8.1", "cuda-version": "13.0.1"}, {"torch-version": "2.8.1", "cuda-version": "13.0"},
{"torch-version": "2.8.0", "cuda-version": "13.0.1"}, {"torch-version": "2.8.0", "cuda-version": "13.0"},
] ]
LINUX_MATRIX = { LINUX_MATRIX = {
@@ -47,28 +47,19 @@ LINUX_MATRIX = {
"2.9.1", "2.9.1",
], ],
"cuda-version": [ "cuda-version": [
"12.4.1", "12.4",
# "12.6.3", # "12.6",
"12.8.1", "12.8",
# "12.9.1", # "12.9",
"13.0.2", "13.0",
], ],
} }
LINUX_ARM64_MATRIX = LINUX_MATRIX LINUX_ARM64_MATRIX = {
LINUX_SELF_HOSTED_MATRIX = {
"flash-attn-version": ["2.7.4"],
"python-version": ["3.10", "3.11", "3.12", "3.13"],
"torch-version": ["2.9.1"],
"cuda-version": ["12.8.1", "13.0.1"],
}
WINDOWS_MATRIX = {
"flash-attn-version": [ "flash-attn-version": [
"2.6.3", "2.6.3",
"2.7.4", # "2.7.4",
"2.8.3", # "2.8.3",
], ],
"python-version": [ "python-version": [
"3.10", "3.10",
@@ -84,11 +75,46 @@ WINDOWS_MATRIX = {
"2.9.1", "2.9.1",
], ],
"cuda-version": [ "cuda-version": [
"12.4.1", "12.4",
"12.6.3", # "12.6",
"12.8.1", "12.8",
# "12.9.1", # "12.9",
"13.0.2", "13.0",
],
}
LINUX_SELF_HOSTED_MATRIX = {
"flash-attn-version": ["2.7.4"],
"python-version": ["3.10", "3.11", "3.12", "3.13"],
"torch-version": ["2.9.1"],
"cuda-version": ["12.8", "13.0"],
}
WINDOWS_MATRIX = {
"flash-attn-version": [
"2.6.3",
# "2.7.4",
# "2.8.3",
],
"python-version": [
"3.10",
"3.11",
"3.12",
# "3.13",
],
"torch-version": [
"2.5.1",
"2.6.0",
"2.7.1",
"2.8.0",
"2.9.1",
],
"cuda-version": [
"12.4",
# "12.6",
"12.8",
# "12.9",
"13.0",
], ],
} }
@@ -96,7 +122,7 @@ WINDOWS_CODEBUILD_MATRIX = {
"flash-attn-version": ["2.6.3", "2.7.4.post1", "2.8.3"], "flash-attn-version": ["2.6.3", "2.7.4.post1", "2.8.3"],
"python-version": ["3.10", "3.11", "3.12", "3.13"], "python-version": ["3.10", "3.11", "3.12", "3.13"],
"torch-version": ["2.9.1"], "torch-version": ["2.9.1"],
"cuda-version": ["13.0.1"], "cuda-version": ["13.0"],
} }