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 PyTorch 2.10 to all matrices and add cu129 for torch 2.9

Browse Source
This commit is contained in:
Mohammad Miadh Angkad
2026-01-15 19:18:58 +08:00
parent 9d7619f3eb
commit c55e718363
2 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
create_matrix.py
+10
View File
@@ -21,12 +21,15 @@ EXCLUDE = [
{"torch-version": "2.7.1", "cuda-version": "12.9"},
# 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-version": "2.5.1", "python-version": "3.14"},
{"torch-version": "2.6.3", "python-version": "3.14"},
@@ -52,6 +55,7 @@ LINUX_MATRIX = {
"2.7.1",
"2.8.0",
"2.9.1",
"2.10.0",
],
"cuda-version": [
"12.4",
@@ -80,6 +84,7 @@ LINUX_ARM64_MATRIX = {
"2.7.1",
# "2.8.0",
"2.9.1",
"2.10.0",
],
"cuda-version": [
"12.4",
@@ -109,6 +114,7 @@ LINUX_SELF_HOSTED_MATRIX = {
"2.7.1",
"2.8.0",
# "2.9.1",
"2.10.0",
],
"cuda-version": [
"12.4",
@@ -138,6 +144,7 @@ LINUX_ARM64_SELF_HOSTED_MATRIX = {
# "2.7.1",
# "2.8.0",
"2.9.1",
"2.10.0",
],
"cuda-version": [
# "12.4",
@@ -166,6 +173,7 @@ WINDOWS_MATRIX = {
# "2.7.1",
# "2.8.0",
"2.9.1",
"2.10.0",
],
"cuda-version": [
# "12.4",
@@ -190,6 +198,7 @@ WINDOWS_CODEBUILD_MATRIX = {
],
"torch-version": [
"2.9.1",
"2.10.0",
],
"cuda-version": [
"12.8",
@@ -215,6 +224,7 @@ WINDOWS_SELF_HOSTED_MATRIX = {
"2.7.1",
"2.8.0",
"2.9.1",
"2.10.0",
],
"cuda-version": [
# "12.4",
get_torch_cuda_version.py
+2 -1
View File
@@ -10,7 +10,8 @@ support_cuda_versions = {
"2.6": [118, 124, 126],
"2.7": [118, 126, 128],
"2.8": [126, 128, 129],
"2.9": [126, 128, 130],
"2.9": [126, 128, 129, 130],
"2.10": [126, 128, 129, 130],
}
cuda_version = int(sys.argv[1])