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: add ARM64 self-hosted runner build support

Browse Source 
- Update build workflow to include ARM64 self-hosted runner job.
- Modify Dockerfile and compose.yml for ARM64 runner setup.
- Add ARM64 self-hosted matrix configuration.
- Update README with ARM64 self-hosted runner instructions.
This commit is contained in:
Junya Morioka
2025-12-11 01:10:52 +09:00
parent 67e64ccae1
commit 521f9dcbaf
6 changed files with 113 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/build.yml
+22
View File
@@ -96,6 +96,26 @@ jobs:
cuda-version: ${{ matrix.cuda-version }}
secrets: inherit
build_wheels_linux_arm64_self_hosted:
name: Build Linux ARM64 (self-hosted)
needs: [create_releases, create_matrix]
if: ${{ fromjson(needs.create_matrix.outputs.matrix).linux_arm64_self_hosted }}
strategy:
fail-fast: false
matrix:
flash-attn-version: ${{ fromjson(needs.create_matrix.outputs.matrix).linux_arm64_self_hosted.flash-attn-version }}
python-version: ${{ fromjson(needs.create_matrix.outputs.matrix).linux_arm64_self_hosted.python-version }}
torch-version: ${{ fromjson(needs.create_matrix.outputs.matrix).linux_arm64_self_hosted.torch-version }}
cuda-version: ${{ fromjson(needs.create_matrix.outputs.matrix).linux_arm64_self_hosted.cuda-version }}
exclude: ${{ fromjson(needs.create_matrix.outputs.matrix).exclude }}
uses: ./.github/workflows/_build_linux_arm_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 }}
secrets: inherit
# #########################################################
# Windows
# #########################################################
@@ -145,6 +165,7 @@ jobs:
- build_wheels_linux
- build_wheels_linux_arm64
- build_wheels_linux_self_hosted
- build_wheels_linux_arm64_self_hosted
- build_wheels_windows
- build_wheels_windows_code_build
if: always()
@@ -173,6 +194,7 @@ jobs:
- build_wheels_linux
- build_wheels_linux_arm64
- build_wheels_linux_self_hosted
- build_wheels_linux_arm64_self_hosted
- build_wheels_windows
- build_wheels_windows_code_build
permissions: