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

ci: refactor wheel path variables and adjust build matrix

Browse Source 
- Replace WHEEL_NAME variables with WHEEL_PATH to store full file paths
  instead of just filenames, eliminating redundant directory concatenation
- Update all workflow references to use WHEEL_PATH directly in install
  and upload commands
- Simplify manylinux wheel handling by storing full paths in output variables
- Reduce tested versions in Linux self-hosted matrix to focus on Python 3.14
- Reduce tested versions in ARM64 self-hosted matrix to focus on Python 3.14
  and flash-attn 2.8.3
- Apply formatting standardization to Windows CodeBuild matrix
This commit is contained in:
Junya Morioka
2025-12-13 17:25:23 +09:00
parent 5d44bd8cd6
commit a7b3fe362e
5 changed files with 77 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files
create_matrix.py
+37 -22
View File
@@ -90,19 +90,20 @@ LINUX_ARM64_MATRIX = {
LINUX_SELF_HOSTED_MATRIX = {
"flash-attn-version": [
"2.7.4",
"2.8.3",
"2.6.3",
# "2.7.4",
# "2.8.3",
],
"python-version": [
"3.10",
"3.11",
"3.12",
"3.13",
# "3.10",
# "3.11",
# "3.12",
# "3.13",
"3.14",
],
"torch-version": ["2.9.1"],
"cuda-version": [
"12.8",
# "12.8",
"13.0",
],
}
@@ -110,26 +111,27 @@ LINUX_SELF_HOSTED_MATRIX = {
LINUX_ARM64_SELF_HOSTED_MATRIX = {
"flash-attn-version": [
# "2.6.3",
"2.7.4",
# "2.8.3",
# "2.7.4",
"2.8.3",
],
"python-version": [
"3.10",
"3.11",
"3.12",
# "3.10",
# "3.11",
# "3.12",
# "3.13",
"3.14",
],
"torch-version": [
"2.5.1",
"2.6.0",
"2.7.1",
# "2.5.1",
# "2.6.0",
# "2.7.1",
# "2.8.0",
"2.9.1",
],
"cuda-version": [
"12.4",
# "12.4",
# "12.6",
"12.8",
# "12.8",
# "12.9",
"13.0",
],
@@ -164,10 +166,23 @@ WINDOWS_MATRIX = {
}
WINDOWS_CODEBUILD_MATRIX = {
"flash-attn-version": ["2.6.3", "2.7.4.post1", "2.8.3"],
"python-version": ["3.10", "3.11", "3.12", "3.13"],
"torch-version": ["2.9.1"],
"cuda-version": ["13.0"],
"flash-attn-version": [
"2.6.3",
"2.7.4.post1",
"2.8.3",
],
"python-version": [
"3.10",
"3.11",
"3.12",
"3.13",
],
"torch-version": [
"2.9.1",
],
"cuda-version": [
"13.0",
],
}
@@ -185,7 +200,7 @@ def main():
"linux_self_hosted": LINUX_SELF_HOSTED_MATRIX,
#
# "linux_arm64_self_hosted": False,
# "linux_arm64_self_hosted": LINUX_ARM64_SELF_HOSTED_MATRIX,
"linux_arm64_self_hosted": LINUX_ARM64_SELF_HOSTED_MATRIX,
#
"windows": False,
# "windows": WINDOWS_MATRIX,