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

Merge pull request #70 from mjun0812/feat/manylinux-build

Browse Source 
feat: add manylinux self-hosted build job and enable version matrix
This commit is contained in:
Junya Morioka
2025-12-17 16:50:56 +09:00
committed by GitHub
parent 8fe2eb90aa acb50f4b17
commit 8bf9488adc
2 changed files with 38 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/build.yml
+26
View File
@@ -35,6 +35,8 @@ jobs:
# #########################################################
# Linux
# #########################################################
# ################ GitHub-hosted runner ################
build_wheels_linux:
name: Build Linux x86_64
needs: [create_releases, create_matrix]
@@ -76,6 +78,7 @@ jobs:
runner: "ubuntu-22.04-arm"
secrets: inherit
# ################ Self-hosted runner ################
build_wheels_linux_self_hosted:
name: Build Linux (self-hosted)
needs: [create_releases, create_matrix]
@@ -118,6 +121,27 @@ jobs:
runner: '["self-hosted-arm", "arm64"]'
secrets: inherit
build_wheels_manylinux_self_hosted:
name: Build manylinux (self-hosted)
needs: [create_releases, create_matrix]
if: ${{ fromjson(needs.create_matrix.outputs.matrix).manylinux_self_hosted }}
strategy:
fail-fast: false
matrix:
flash-attn-version: ${{ fromjson(needs.create_matrix.outputs.matrix).manylinux_self_hosted.flash-attn-version }}
python-version: ${{ fromjson(needs.create_matrix.outputs.matrix).manylinux_self_hosted.python-version }}
torch-version: ${{ fromjson(needs.create_matrix.outputs.matrix).manylinux_self_hosted.torch-version }}
cuda-version: ${{ fromjson(needs.create_matrix.outputs.matrix).manylinux_self_hosted.cuda-version }}
exclude: ${{ fromjson(needs.create_matrix.outputs.matrix).exclude }}
uses: ./.github/workflows/_build_manylinux_self_host.yml
with:
flash-attn-version: ${{ matrix.flash-attn-version }}
python-version: ${{ matrix.python-version }}
torch-version: ${{ matrix.torch-version }}
cuda-version: ${{ matrix.cuda-version }}
runner: '["self-hosted", "x64"]'
secrets: inherit
# #########################################################
# Windows
# #########################################################
@@ -168,6 +192,7 @@ jobs:
- build_wheels_linux_arm64
- build_wheels_linux_self_hosted
- build_wheels_linux_arm64_self_hosted
- build_wheels_manylinux_self_hosted
- build_wheels_windows
- build_wheels_windows_code_build
if: always()
@@ -197,6 +222,7 @@ jobs:
- build_wheels_linux_arm64
- build_wheels_linux_self_hosted
- build_wheels_linux_arm64_self_hosted
- build_wheels_manylinux_self_hosted
- build_wheels_windows
- build_wheels_windows_code_build
permissions:
create_matrix.py
+12 -9
View File
@@ -91,19 +91,19 @@ LINUX_ARM64_MATRIX = {
LINUX_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",
# "3.14",
],
"torch-version": ["2.9.1"],
"cuda-version": [
# "12.8",
"12.8",
"13.0",
],
}
@@ -200,11 +200,14 @@ def main():
"linux_self_hosted": False,
# "linux_self_hosted": LINUX_SELF_HOSTED_MATRIX,
#
# "manylinux_self_hosted": False,
"manylinux_self_hosted": LINUX_SELF_HOSTED_MATRIX,
#
"linux_arm64_self_hosted": False,
# "linux_arm64_self_hosted": LINUX_ARM64_SELF_HOSTED_MATRIX,
#
# "windows": False,
"windows": WINDOWS_MATRIX,
"windows": False,
# "windows": WINDOWS_MATRIX,
#
"windows_code_build": False,
# "windows_code_build": WINDOWS_CODEBUILD_MATRIX,