yindo/flash-attention-prebuild-wheels-rocm
1
0
Fork 0
mirror of https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git synced 2026-07-01 01:37:53 -04:00
Code Issues Packages Projects Releases 2 Wiki Activity

fix self hosted runner

Browse Source
This commit is contained in:
Junya Morioka
2025-05-17 14:41:51 +09:00
parent abee9c0ecc
commit b2c3d05e7b
3 changed files with 14 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/test-self-hosted.yml
+1 -5
View File
@@ -7,8 +7,6 @@ jobs:
runs-on: runs-on:
- self-hosted - self-hosted
timeout-minutes: 10 timeout-minutes: 10
container:
image: ghcr.io/actions/actions-runner:latest
steps: steps:
- run: ls - run: ls
- run: pwd - run: pwd
@@ -20,13 +18,11 @@ jobs:
strategy: strategy:
fail-fast: false fail-fast: false
matrix: matrix:
flash-attn-version: ["2.4.3"] flash-attn-version: ["2.4.3", "2.5.9"]
python-version: ["3.12"] python-version: ["3.12"]
torch-version: ["2.7.0"] torch-version: ["2.7.0"]
# https://developer.nvidia.com/cuda-toolkit-archive # https://developer.nvidia.com/cuda-toolkit-archive
cuda-version: ["12.8.1"] cuda-version: ["12.8.1"]
container:
image: ghcr.io/actions/actions-runner
steps: steps:
- uses: actions/checkout@v4 - uses: actions/checkout@v4
- uses: actions/setup-python@v5 - uses: actions/setup-python@v5
self-hosted-runner/Dockerfile
+11 -7
View File
@@ -1,4 +1,4 @@
FROM ubuntu:latest FROM ubuntu:24.04
ARG REPOSITORY_URL ARG REPOSITORY_URL
ARG PERSONAL_ACCESS_TOKEN ARG PERSONAL_ACCESS_TOKEN
@@ -29,20 +29,24 @@ RUN install -m 0755 -d /etc/apt/keyrings \
&& apt-get update \ && apt-get update \
&& apt-get install -y --no-install-recommends \ && apt-get install -y --no-install-recommends \
docker-ce-cli \ docker-ce-cli \
# docker-ce \
# containerd.io \
# docker-buildx-plugin \
# docker-compose-plugin \
&& rm -rf /var/lib/apt/lists/* && rm -rf /var/lib/apt/lists/*
# Install dotnet # Install dotnet
RUN add-apt-repository ppa:dotnet/backports RUN add-apt-repository ppa:dotnet/backports
# Create runner user and setup group # ubuntu >= 24.04 image has user named 'ubuntu'
RUN useradd -mr -d /home/runner runner \ RUN usermod -aG sudo ubuntu \
&& usermod -aG sudo runner && usermod -aG docker ubuntu
# Allow sudo without password # Allow sudo without password
RUN echo '%sudo ALL=(ALL) NOPASSWD: ALL' >> /etc/sudoers RUN echo '%sudo ALL=(ALL) NOPASSWD: ALL' >> /etc/sudoers
USER runner USER ubuntu
WORKDIR /home/runner WORKDIR /home/ubuntu
RUN curl -fsSL -o actions-runner.tar.gz -L "https://github.com/actions/runner/releases/download/v${GH_RUNNER_VERSION}/actions-runner-linux-${TARGET_ARCH}-${GH_RUNNER_VERSION}.tar.gz" \ RUN curl -fsSL -o actions-runner.tar.gz -L "https://github.com/actions/runner/releases/download/v${GH_RUNNER_VERSION}/actions-runner-linux-${TARGET_ARCH}-${GH_RUNNER_VERSION}.tar.gz" \
&& tar xf actions-runner.tar.gz \ && tar xf actions-runner.tar.gz \
@@ -55,7 +59,7 @@ RUN curl -fsSL -o actions-runner.tar.gz -L "https://github.com/actions/runner/re
--name $RUNNER_NAME \ --name $RUNNER_NAME \
--runnergroup $RUNNER_GROUP \ --runnergroup $RUNNER_GROUP \
--labels "${RUNNER_LABELS},${TARGET_ARCH}" \ --labels "${RUNNER_LABELS},${TARGET_ARCH}" \
--work /home/runner/actions-runner \ --work /home/ubuntu/actions-runner \
--replace --replace
CMD ["./run.sh"] CMD ["./run.sh"]
self-hosted-runner/compose.yml
+2 -1
View File
@@ -4,8 +4,9 @@ services:
context: . context: .
dockerfile: Dockerfile dockerfile: Dockerfile
args: args:
PERSONAL_ACCESS_TOKEN: $PERSONAL_ACCESS_TOKEN
REPOSITORY_URL: https://github.com/mjun0812/flash-attention-prebuild-wheels REPOSITORY_URL: https://github.com/mjun0812/flash-attention-prebuild-wheels
PERSONAL_ACCESS_TOKEN: $PERSONAL_ACCESS_TOKEN
GH_RUNNER_VERSION: 2.324.0
RUNNER_NAME: self-hosted-runner RUNNER_NAME: self-hosted-runner
RUNNER_GROUP: default RUNNER_GROUP: default
RUNNER_LABELS: self-hosted RUNNER_LABELS: self-hosted