mirror of
https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git
synced 2026-07-01 01:37:53 -04:00
refactor: reorganize auditwheel repair and add manylinux platform support
- Move auditwheel repair after initial release upload with continue-on-error to allow pipeline continuation - Add manylinux platform normalization support in normalize_platform_name() - Expand self-hosted build matrix to include Python 3.14 and Flash Attention 2.8.3 - Improve wheel upload flow by separating regular and manylinux wheel handling
This commit is contained in:
+18
-6
@@ -89,10 +89,22 @@ LINUX_ARM64_MATRIX = {
|
||||
}
|
||||
|
||||
LINUX_SELF_HOSTED_MATRIX = {
|
||||
"flash-attn-version": ["2.7.4"],
|
||||
"python-version": ["3.10", "3.11", "3.12", "3.13"],
|
||||
"flash-attn-version": [
|
||||
"2.7.4",
|
||||
"2.8.3",
|
||||
],
|
||||
"python-version": [
|
||||
"3.10",
|
||||
"3.11",
|
||||
"3.12",
|
||||
"3.13",
|
||||
"3.14",
|
||||
],
|
||||
"torch-version": ["2.9.1"],
|
||||
"cuda-version": ["12.8", "13.0"],
|
||||
"cuda-version": [
|
||||
"12.8",
|
||||
"13.0",
|
||||
],
|
||||
}
|
||||
|
||||
LINUX_ARM64_SELF_HOSTED_MATRIX = {
|
||||
@@ -169,11 +181,11 @@ def main():
|
||||
"linux_arm64": False,
|
||||
# "linux_arm64": LINUX_ARM64_MATRIX,
|
||||
#
|
||||
"linux_self_hosted": False,
|
||||
# "linux_self_hosted": LINUX_SELF_HOSTED_MATRIX,
|
||||
# "linux_self_hosted": False,
|
||||
"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,
|
||||
|
||||
Reference in New Issue
Block a user