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

feat: update create_matrix.py for windows

Browse Source
This commit is contained in:
Junya Morioka
2025-12-16 20:52:14 +09:00
parent 65a4cd54b3
commit 5bc98e4b5d
1 changed files with 13 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
create_matrix.py
+13 -12
View File
@@ -167,21 +167,22 @@ WINDOWS_MATRIX = {
WINDOWS_CODEBUILD_MATRIX = {
"flash-attn-version": [
"2.6.3",
"2.7.4.post1",
# "2.6.3",
# "2.7.4.post1",
"2.8.3",
],
"python-version": [
"3.10",
"3.11",
# "3.10",
# "3.11",
"3.12",
"3.13",
# "3.13",
],
"torch-version": [
"2.9.1",
],
"cuda-version": [
"13.0",
"12.8",
# "13.0",
],
}
@@ -196,14 +197,14 @@ 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,
"linux_arm64_self_hosted": False,
# "linux_arm64_self_hosted": LINUX_ARM64_SELF_HOSTED_MATRIX,
#
"windows": False,
# "windows": WINDOWS_MATRIX,
# "windows": False,
"windows": WINDOWS_MATRIX,
#
"windows_code_build": False,
# "windows_code_build": WINDOWS_CODEBUILD_MATRIX,