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 flash_attn 2.7 and PyTorch 2.9

Browse Source
This commit is contained in:
Junya Morioka
2025-11-04 20:19:53 +09:00
parent 18d910e698
commit e1a4c88fac
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
create_matrix.py
+6 -6
View File
@@ -35,10 +35,10 @@ LINUX_MATRIX = {
} }
LINUX_SELF_HOSTED_MATRIX = { LINUX_SELF_HOSTED_MATRIX = {
"flash-attn-version": ["2.7.4.post1", "2.8.3"], "flash-attn-version": ["2.7.4"],
"python-version": ["3.10", "3.11", "3.12", "3.13"], "python-version": ["3.10", "3.11", "3.12", "3.13"],
"torch-version": ["2.9.0"], "torch-version": ["2.9.0"],
"cuda-version": ["13.0.1"], "cuda-version": ["12.8.1", "13.0.1"],
} }
WINDOWS_MATRIX = { WINDOWS_MATRIX = {
@@ -62,10 +62,10 @@ def main():
{ {
# "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,
"windows_code_build": False, "windows_code_build": False,
"exclude": EXCLUDE, "exclude": EXCLUDE,