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 building for arm64 and openci runner test

Browse Source
This commit is contained in:
Junya Morioka
2025-12-05 01:15:34 +09:00
parent bdaa0f9bc5
commit d1715633db
8 changed files with 120 additions and 162 deletions
Download Patch File Download Diff File Expand all files Collapse all files
create_matrix.py
+23 -8
View File
@@ -19,6 +19,12 @@ EXCLUDE = [
{"torch-version": "2.7.1", "cuda-version": "12.9.1"},
# torch >= 2.9 does not support Python 3.9
{"torch-version": "2.9.1", "python-version": "3.9"},
# torch < 2.9 does not support CUDA 13.0
{"torch-version": "2.5.1", "cuda-version": "13.0.1"},
{"torch-version": "2.6.0", "cuda-version": "13.0.1"},
{"torch-version": "2.7.1", "cuda-version": "13.0.1"},
{"torch-version": "2.8.1", "cuda-version": "13.0.1"},
{"torch-version": "2.8.0", "cuda-version": "13.0.1"},
]
LINUX_MATRIX = {
@@ -31,7 +37,7 @@ LINUX_MATRIX = {
"3.10",
"3.11",
"3.12",
# "3.13"
"3.13",
],
"torch-version": [
"2.5.1",
@@ -42,13 +48,15 @@ LINUX_MATRIX = {
],
"cuda-version": [
"12.4.1",
"12.6.3",
# "12.6.3",
"12.8.1",
# "12.9.1",
"13.0.2",
],
}
LINUX_ARM64_MATRIX = LINUX_MATRIX
LINUX_SELF_HOSTED_MATRIX = {
"flash-attn-version": ["2.7.4"],
"python-version": ["3.10", "3.11", "3.12", "3.13"],
@@ -96,14 +104,21 @@ def main():
print(
json.dumps(
{
"linux": LINUX_MATRIX,
# "linux": False,
# "linux_self_hosted": LINUX_SELF_HOSTED_MATRIX,
"linux": False,
# "linux": LINUX_MATRIX,
#
# "linux_arm64": False,
"linux_arm64": LINUX_ARM64_MATRIX,
#
"linux_self_hosted": False,
"windows": WINDOWS_MATRIX,
# "windows": False,
# "windows_code_build": WINDOWS_CODEBUILD_MATRIX,
# "linux_self_hosted": LINUX_SELF_HOSTED_MATRIX,
#
"windows": False,
# "windows": WINDOWS_MATRIX,
#
"windows_code_build": False,
# "windows_code_build": WINDOWS_CODEBUILD_MATRIX,
#
"exclude": EXCLUDE,
}
)