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

rocm only

Browse Source
This commit is contained in:
John Doe
2026-02-10 20:53:09 -05:00
parent 6ac2a69fa9
commit 2332e6fcae
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
create_matrix.py
+7 -7
View File
@@ -275,25 +275,25 @@ def main():
print( print(
json.dumps( json.dumps(
{ {
"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,
# #
"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,
# #
#"rocm": False, #"rocm": False,