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: add missing wheel build matrices for Linux x86_64 and ARM64

Browse Source 
- Expand LINUX_SELF_HOSTED_MATRIX with torch 2.6.0-2.9.1 and CUDA
  12.4/12.8/13.0 to cover 29 missing Linux x86_64 packages
- Add torch 2.9.1 to LINUX_ARM64_MATRIX for 9 missing ARM64 packages
- Enable linux_arm64 matrix in main() output
This commit is contained in:
Junya Morioka
2026-01-28 18:17:12 +09:00
parent ee06a9be5e
commit e9f16ebc01
1 changed files with 10 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
create_matrix.py
+10 -11
View File
@@ -91,7 +91,7 @@ LINUX_ARM64_MATRIX = {
# "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": [
@@ -118,19 +118,18 @@ LINUX_SELF_HOSTED_MATRIX = {
], ],
"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",
# "13.1",
], ],
} }
@@ -254,8 +253,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,