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:27:54 -04:00
Code Issues Packages Projects Releases 2 Wiki Activity

chore: Update create_matrix.py

Browse Source
This commit is contained in:
Junya Morioka
2025-12-21 11:22:06 +09:00
parent 6502b392f9
commit 89bfe94ab1
1 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
create_matrix.py
+13 -4
View File
@@ -98,12 +98,21 @@ LINUX_SELF_HOSTED_MATRIX = {
"3.10",
"3.11",
"3.12",
# "3.13",
"3.13",
# "3.14",
],
"torch-version": ["2.9.1"],
"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",
],
}
@@ -197,8 +206,8 @@ def main():
"linux_arm64": False,
# "linux_arm64": LINUX_ARM64_MATRIX,
#
"linux_self_hosted": False,
# "linux_self_hosted": LINUX_SELF_HOSTED_MATRIX,
# "linux_self_hosted": False,
"linux_self_hosted": LINUX_SELF_HOSTED_MATRIX,
#
"linux_arm64_self_hosted": False,
# "linux_arm64_self_hosted": LINUX_ARM64_SELF_HOSTED_MATRIX,