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

docs: update README.md

Browse Source
This commit is contained in:
Junya Morioka
2025-11-05 14:54:11 +09:00
parent d2ebd07628
commit 3ac3b409a4
2 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+1
View File
@@ -3340,3 +3340,4 @@ docker compose up -d
year={2024} year={2024}
} }
``` ```
create_matrix.py
+9 -5
View File
@@ -22,7 +22,10 @@ EXCLUDE = [
] ]
LINUX_MATRIX = { LINUX_MATRIX = {
"flash-attn-version": ["2.6.3", "2.7.4.post1", "2.8.3"], "flash-attn-version": [
# "2.6.3", "2.7.4.post1", "2.8.3"
"2.8.1"
],
"python-version": ["3.10", "3.11", "3.12", "3.13"], "python-version": ["3.10", "3.11", "3.12", "3.13"],
"torch-version": [ "torch-version": [
# "2.5.1", "2.6.0", "2.7.1", "2.8.0", # "2.5.1", "2.6.0", "2.7.1", "2.8.0",
@@ -30,6 +33,7 @@ LINUX_MATRIX = {
], ],
"cuda-version": [ "cuda-version": [
# "12.4.1", "12.6.3", "12.8.1", "12.9.1", # "12.4.1", "12.6.3", "12.8.1", "12.9.1",
"12.8.1",
"13.0.1", "13.0.1",
], ],
} }
@@ -60,10 +64,10 @@ def main():
print( print(
json.dumps( json.dumps(
{ {
# "linux": LINUX_MATRIX, "linux": LINUX_MATRIX,
"linux": False, # "linux": False,
"linux_self_hosted": LINUX_SELF_HOSTED_MATRIX, # "linux_self_hosted": LINUX_SELF_HOSTED_MATRIX,
# "linux_self_hosted": False, "linux_self_hosted": False,
# "windows": WINDOWS_MATRIX, # "windows": WINDOWS_MATRIX,
"windows": False, "windows": False,
# "windows_code_build": WINDOWS_CODEBUILD_MATRIX, # "windows_code_build": WINDOWS_CODEBUILD_MATRIX,