mirror of
https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git
synced 2026-07-01 01:27:54 -04:00
Junya Morioka
5d44bd8cd6
- Replace actions/setup-python with uv-based Python installation - Consolidate build dependencies into Python installation step - Enable auditwheel repair and manylinux wheel generation across all Linux builds - Add patchelf as build dependency for glibc compatibility - Update Python version to 3.14 in build and test workflows
80 lines
2.9 KiB
YAML
80 lines
2.9 KiB
YAML
name: Test dynamic matrix
|
|
|
|
on:
|
|
workflow_dispatch:
|
|
|
|
jobs:
|
|
create_matrix:
|
|
name: Create Matrix
|
|
runs-on: ubuntu-latest
|
|
outputs:
|
|
matrix: ${{ steps.create_matrix.outputs.matrix }}
|
|
steps:
|
|
- uses: actions/checkout@v4
|
|
- name: Create Matrix
|
|
id: create_matrix
|
|
run: |
|
|
python create_matrix.py | tee /tmp/matrix.json
|
|
echo "matrix=$(cat /tmp/matrix.json)" >> $GITHUB_OUTPUT
|
|
|
|
# #########################################################
|
|
# Linux
|
|
# #########################################################
|
|
build_wheels_linux:
|
|
name: Build Linux
|
|
runs-on: ubuntu-latest
|
|
needs: [create_matrix]
|
|
if: ${{ fromjson(needs.create_matrix.outputs.matrix).linux }}
|
|
strategy:
|
|
fail-fast: false
|
|
matrix:
|
|
flash-attn-version: ${{ fromjson(needs.create_matrix.outputs.matrix).linux.flash-attn-version }}
|
|
python-version: ${{ fromjson(needs.create_matrix.outputs.matrix).linux.python-version }}
|
|
torch-version: ${{ fromjson(needs.create_matrix.outputs.matrix).linux.torch-version }}
|
|
cuda-version: ${{ fromjson(needs.create_matrix.outputs.matrix).linux.cuda-version }}
|
|
exclude: ${{ fromjson(needs.create_matrix.outputs.matrix).exclude }}
|
|
steps:
|
|
- name: Test Linux
|
|
run: |
|
|
echo "Flash Attn Version: ${{ matrix.flash-attn-version }}"
|
|
echo "Python Version: ${{ matrix.python-version }}"
|
|
echo "Torch Version: ${{ matrix.torch-version }}"
|
|
echo "CUDA Version: ${{ matrix.cuda-version }}"
|
|
|
|
build_wheels_linux_self_hosted:
|
|
name: Build Linux (self-hosted)
|
|
runs-on: ubuntu-latest
|
|
needs: [create_matrix]
|
|
if: ${{ fromjson(needs.create_matrix.outputs.matrix).linux_self_hosted }}
|
|
strategy:
|
|
fail-fast: false
|
|
matrix:
|
|
flash-attn-version: ${{ fromjson(needs.create_matrix.outputs.matrix).linux_self_hosted.flash-attn-version }}
|
|
python-version: ${{ fromjson(needs.create_matrix.outputs.matrix).linux_self_hosted.python-version }}
|
|
torch-version: ${{ fromjson(needs.create_matrix.outputs.matrix).linux_self_hosted.torch-version }}
|
|
cuda-version: ${{ fromjson(needs.create_matrix.outputs.matrix).linux_self_hosted.cuda-version }}
|
|
exclude: ${{ fromjson(needs.create_matrix.outputs.matrix).exclude }}
|
|
steps:
|
|
- name: Test Linux (self-hosted)
|
|
run: |
|
|
echo "Flash Attn Version: ${{ matrix.flash-attn-version }}"
|
|
echo "Python Version: ${{ matrix.python-version }}"
|
|
echo "Torch Version: ${{ matrix.torch-version }}"
|
|
echo "CUDA Version: ${{ matrix.cuda-version }}"
|
|
|
|
update_release_notes:
|
|
name: Update Release Notes
|
|
needs:
|
|
- build_wheels_linux
|
|
- build_wheels_linux_self_hosted
|
|
permissions:
|
|
contents: write
|
|
if: always()
|
|
runs-on: ubuntu-latest
|
|
steps:
|
|
- uses: actions/checkout@v4
|
|
|
|
- uses: actions/setup-python@v5
|
|
with:
|
|
python-version: 3.14
|