mirror of
https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git
synced 2026-07-01 01:37:53 -04:00
Junya Morioka
521f9dcbaf
- 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.
46 lines
1.0 KiB
YAML
46 lines
1.0 KiB
YAML
services:
|
|
runner:
|
|
privileged: true
|
|
restart: always
|
|
env_file:
|
|
- .env
|
|
environment:
|
|
REPOSITORY_URL: https://github.com/mjun0812/flash-attention-prebuild-wheels
|
|
RUNNER_NAME: self-hosted-runner
|
|
RUNNER_GROUP: default
|
|
TARGET_ARCH: x64
|
|
build:
|
|
context: .
|
|
dockerfile: Dockerfile
|
|
args:
|
|
GH_RUNNER_VERSION: 2.329.0
|
|
TARGET_ARCH: x64
|
|
PLATFORM: linux/amd64
|
|
volumes:
|
|
- fa-self:/var/lib/docker
|
|
|
|
runner-arm:
|
|
platform: linux/arm64
|
|
privileged: true
|
|
restart: always
|
|
env_file:
|
|
- .env.arm
|
|
environment:
|
|
REPOSITORY_URL: https://github.com/mjun0812/flash-attention-prebuild-wheels
|
|
RUNNER_NAME: self-hosted-runner
|
|
RUNNER_GROUP: default
|
|
TARGET_ARCH: arm64
|
|
build:
|
|
context: .
|
|
dockerfile: Dockerfile
|
|
args:
|
|
GH_RUNNER_VERSION: 2.329.0
|
|
TARGET_ARCH: arm64
|
|
PLATFORM: linux/arm64
|
|
volumes:
|
|
- fa-self-arm:/var/lib/docker
|
|
|
|
volumes:
|
|
fa-self:
|
|
fa-self-arm:
|