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: Update arm matrix

Browse Source
This commit is contained in:
Junya Morioka
2025-12-06 15:04:51 +09:00
parent eda3d159bd
commit 5889fcc8d7
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
create_matrix.py
+4 -4
View File
@@ -57,9 +57,9 @@ LINUX_MATRIX = {
LINUX_ARM64_MATRIX = { LINUX_ARM64_MATRIX = {
"flash-attn-version": [ "flash-attn-version": [
"2.6.3", # "2.6.3",
# "2.7.4", "2.7.4",
# "2.8.3", "2.8.3",
], ],
"python-version": [ "python-version": [
"3.10", "3.10",
@@ -71,7 +71,7 @@ LINUX_ARM64_MATRIX = {
"2.5.1", "2.5.1",
"2.6.0", "2.6.0",
"2.7.1", "2.7.1",
"2.8.0", # "2.8.0",
"2.9.1", "2.9.1",
], ],
"cuda-version": [ "cuda-version": [