mirror of
https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git
synced 2026-07-01 01:37:53 -04:00
fix: test workflow
This commit is contained in:
@@ -42,6 +42,7 @@ jobs:
|
||||
|
||||
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:
|
||||
@@ -52,13 +53,13 @@ jobs:
|
||||
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 }}
|
||||
uses: ./.github/workflows/build_linux_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
|
||||
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
|
||||
|
||||
Reference in New Issue
Block a user