mirror of
https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git
synced 2026-07-01 01:27:54 -04:00
Junya Morioka
2f3b1245f7
- Merge _build_manylinux_self_host.yml into _build_linux_self_host.yml - Add automatic package manager detection (apt-get/dnf) for both x86_64 and ARM64 - Add environment check steps to all self-hosted workflows - Update build.yml to use unified workflow with container-image parameter - Remove duplicate build_wheels_manylinux_self_hosted job - Update test workflows to use consolidated workflow - Remove manylinux_self_hosted matrix configuration - Set manylinux container image for ARM64 test workflow
30 lines
1018 B
YAML
30 lines
1018 B
YAML
name: Test ARM Linux build (self-hosted)
|
|
|
|
on:
|
|
workflow_dispatch:
|
|
|
|
jobs:
|
|
# #########################################################
|
|
# Build wheels with self-hosted runner
|
|
# #########################################################
|
|
build_wheels_self_hosted:
|
|
name: Build wheels and Upload (Linux ARM64, self-hosted runner)
|
|
uses: ./.github/workflows/_build_linux_arm_self_host.yml
|
|
strategy:
|
|
fail-fast: false
|
|
matrix:
|
|
flash-attn-version: ["2.8.3"]
|
|
python-version: ["3.13"]
|
|
torch-version: ["2.9.1"]
|
|
# https://developer.nvidia.com/cuda-toolkit-archive
|
|
cuda-version: ["13.0.1"]
|
|
runner: ['["self-hosted-arm"]']
|
|
with:
|
|
flash-attn-version: ${{ matrix.flash-attn-version }}
|
|
python-version: ${{ matrix.python-version }}
|
|
torch-version: ${{ matrix.torch-version }}
|
|
cuda-version: ${{ matrix.cuda-version }}
|
|
is-upload: false
|
|
runner: ${{ matrix.runner}}
|
|
container-image: "quay.io/pypa/manylinux_2_28_aarch64"
|