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

build: update build matrix configurations

Browse Source 
- Update CUDA versions in Linux self-hosted matrix
- Update Python, PyTorch, and CUDA versions in Windows self-hosted matrix
- Enable Windows self-hosted build and disable Linux ARM64 and Windows standard builds
This commit is contained in:
Junya Morioka
2026-01-27 18:41:33 +09:00
parent 6e93ad0663
commit 28773fb7dc
1 changed files with 15 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
create_matrix.py
+15 -14
View File
@@ -127,9 +127,9 @@ LINUX_SELF_HOSTED_MATRIX = {
"cuda-version": [ "cuda-version": [
# "12.4", # "12.4",
"12.6", "12.6",
"12.8", # "12.8",
# "12.9", # "12.9",
"13.0", # "13.0",
# "13.1", # "13.1",
], ],
} }
@@ -227,21 +227,22 @@ WINDOWS_SELF_HOSTED_MATRIX = {
"3.11", "3.11",
"3.12", "3.12",
"3.13", "3.13",
"3.14",
], ],
"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",
# "2.10.0", "2.10.0",
], ],
"cuda-version": [ "cuda-version": [
# "12.4", # "12.4",
# "12.6", # "12.6",
"12.8", "12.8",
# "12.9", # "12.9",
# "13.0", "13.0",
], ],
} }
@@ -253,8 +254,8 @@ def main():
"linux": False, "linux": False,
# "linux": LINUX_MATRIX, # "linux": LINUX_MATRIX,
# #
# "linux_arm64": False, "linux_arm64": False,
"linux_arm64": LINUX_ARM64_MATRIX, # "linux_arm64": LINUX_ARM64_MATRIX,
# #
# "linux_self_hosted": False, # "linux_self_hosted": False,
"linux_self_hosted": LINUX_SELF_HOSTED_MATRIX, "linux_self_hosted": LINUX_SELF_HOSTED_MATRIX,
@@ -262,11 +263,11 @@ def main():
"linux_arm64_self_hosted": False, "linux_arm64_self_hosted": False,
# "linux_arm64_self_hosted": LINUX_ARM64_SELF_HOSTED_MATRIX, # "linux_arm64_self_hosted": LINUX_ARM64_SELF_HOSTED_MATRIX,
# #
# "windows": False, "windows": False,
"windows": WINDOWS_MATRIX, # "windows": WINDOWS_MATRIX,
# #
"windows_self_hosted": False, # "windows_self_hosted": False,
# "windows_self_hosted": WINDOWS_SELF_HOSTED_MATRIX, "windows_self_hosted": WINDOWS_SELF_HOSTED_MATRIX,
# #
"windows_code_build": False, "windows_code_build": False,
# "windows_code_build": WINDOWS_CODEBUILD_MATRIX, # "windows_code_build": WINDOWS_CODEBUILD_MATRIX,