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

Merge pull request #72 from mmangkad/add-torch-2.10-cu129

Browse Source 
feat: Add PyTorch 2.10 to all matrices; enable cu129 for torch 2.9
This commit is contained in:
Junya Morioka
2026-01-17 23:53:56 +09:00
committed by GitHub
parent 9c9da312ac c55e718363
commit 41cd5400fb
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-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-version": "2.10.0", "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"}, {"torch-version": "2.5.1", "cuda-version": "13.0"},
{"torch-version": "2.6.0", "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.7.1", "cuda-version": "13.0"},
{"torch-version": "2.8.1", "cuda-version": "13.0"}, {"torch-version": "2.8.1", "cuda-version": "13.0"},
{"torch-version": "2.8.0", "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 # Python 3.14 is supported from torch 2.9
{"torch-version": "2.5.1", "python-version": "3.14"}, {"torch-version": "2.5.1", "python-version": "3.14"},
{"torch-version": "2.6.3", "python-version": "3.14"}, {"torch-version": "2.6.3", "python-version": "3.14"},
@@ -52,6 +55,7 @@ LINUX_MATRIX = {
"2.7.1", "2.7.1",
"2.8.0", "2.8.0",
"2.9.1", "2.9.1",
"2.10.0",
], ],
"cuda-version": [ "cuda-version": [
"12.4", "12.4",
@@ -80,6 +84,7 @@ LINUX_ARM64_MATRIX = {
"2.7.1", "2.7.1",
# "2.8.0", # "2.8.0",
"2.9.1", "2.9.1",
"2.10.0",
], ],
"cuda-version": [ "cuda-version": [
"12.4", "12.4",
@@ -109,6 +114,7 @@ LINUX_SELF_HOSTED_MATRIX = {
"2.7.1", "2.7.1",
"2.8.0", "2.8.0",
# "2.9.1", # "2.9.1",
"2.10.0",
], ],
"cuda-version": [ "cuda-version": [
"12.4", "12.4",
@@ -138,6 +144,7 @@ LINUX_ARM64_SELF_HOSTED_MATRIX = {
# "2.7.1", # "2.7.1",
# "2.8.0", # "2.8.0",
"2.9.1", "2.9.1",
"2.10.0",
], ],
"cuda-version": [ "cuda-version": [
# "12.4", # "12.4",
@@ -166,6 +173,7 @@ WINDOWS_MATRIX = {
# "2.7.1", # "2.7.1",
# "2.8.0", # "2.8.0",
"2.9.1", "2.9.1",
"2.10.0",
], ],
"cuda-version": [ "cuda-version": [
# "12.4", # "12.4",
@@ -190,6 +198,7 @@ WINDOWS_CODEBUILD_MATRIX = {
], ],
"torch-version": [ "torch-version": [
"2.9.1", "2.9.1",
"2.10.0",
], ],
"cuda-version": [ "cuda-version": [
"12.8", "12.8",
@@ -215,6 +224,7 @@ WINDOWS_SELF_HOSTED_MATRIX = {
"2.7.1", "2.7.1",
"2.8.0", "2.8.0",
"2.9.1", "2.9.1",
"2.10.0",
], ],
"cuda-version": [ "cuda-version": [
# "12.4", # "12.4",
get_torch_cuda_version.py
+2 -1
View File
@@ -10,7 +10,8 @@ support_cuda_versions = {
"2.6": [118, 124, 126], "2.6": [118, 124, 126],
"2.7": [118, 126, 128], "2.7": [118, 126, 128],
"2.8": [126, 128, 129], "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]) cuda_version = int(sys.argv[1])