mirror of
https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git
synced 2026-07-01 01:37:53 -04:00
ci: add architecture labels to self-hosted runner configuration
- Add arm64 label to ARM64 self-hosted runner job - Add x64 label to x86_64 self-hosted runner jobs (both container and non-container) - Ensures runner selection by both custom runner name and architecture type
This commit is contained in:
@@ -42,7 +42,9 @@ on:
|
||||
jobs:
|
||||
build_wheels_self_hosted:
|
||||
name: Build wheels and Upload (Linux x86_64, self-hosted runner)
|
||||
runs-on: ${{ inputs.runner }}
|
||||
runs-on:
|
||||
- ${{ inputs.runner }}
|
||||
- arm64
|
||||
container:
|
||||
image: ${{ inputs.container-image }}
|
||||
options: --platform linux/arm64
|
||||
|
||||
@@ -48,7 +48,9 @@ jobs:
|
||||
build_wheels_self_hosted:
|
||||
if: ${{ inputs.use-container }}
|
||||
name: Build wheels and Upload (Linux x86_64, self-hosted runner)
|
||||
runs-on: ${{ inputs.runner }}
|
||||
runs-on:
|
||||
- ${{ inputs.runner }}
|
||||
- x64
|
||||
container:
|
||||
image: ${{ inputs.container-image }}
|
||||
defaults:
|
||||
@@ -185,7 +187,9 @@ jobs:
|
||||
build_wheels_self_hosted_no_container:
|
||||
if: ${{ !inputs.use-container }}
|
||||
name: Build wheels and Upload (Linux x86_64, self-hosted runner)
|
||||
runs-on: ${{ inputs.runner }}
|
||||
runs-on:
|
||||
- ${{ inputs.runner }}
|
||||
- x64
|
||||
defaults:
|
||||
run:
|
||||
shell: bash
|
||||
|
||||
Reference in New Issue
Block a user