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

fix: reduce jobs

Browse Source
This commit is contained in:
Junya Morioka
2025-11-27 12:38:03 +09:00
parent 814e0b4067
commit 726843716a
1 changed files with 14 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
create_matrix.py
+14 -5
View File
@@ -22,8 +22,17 @@ EXCLUDE = [
]
LINUX_MATRIX = {
"flash-attn-version": ["2.6.3", "2.7.4", "2.8.3"],
"python-version": ["3.10", "3.11", "3.12", "3.13"],
"flash-attn-version": [
"2.6.3",
"2.7.4",
"2.8.3",
],
"python-version": [
"3.10",
"3.11",
"3.12",
# "3.13"
],
"torch-version": [
"2.5.1",
"2.6.0",
@@ -35,7 +44,7 @@ LINUX_MATRIX = {
"12.4.1",
"12.6.3",
"12.8.1",
"12.9.1",
# "12.9.1",
"13.0.2",
],
}
@@ -57,7 +66,7 @@ WINDOWS_MATRIX = {
"3.10",
"3.11",
"3.12",
"3.13",
# "3.13",
],
"torch-version": [
"2.5.1",
@@ -70,7 +79,7 @@ WINDOWS_MATRIX = {
"12.4.1",
"12.6.3",
"12.8.1",
"12.9.1",
# "12.9.1",
"13.0.2",
],
}