mirror of
https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git
synced 2026-07-01 01:37:53 -04:00
Merge pull request #8 from mjun0812/feature/add-self-hosted-runner
[WIP] update test workflow
This commit is contained in:
@@ -25,6 +25,8 @@ jobs:
|
||||
runs-on:
|
||||
- self-hosted
|
||||
timeout-minutes: 1000
|
||||
container:
|
||||
image: ubuntu:22.04
|
||||
strategy:
|
||||
fail-fast: false
|
||||
matrix:
|
||||
@@ -34,6 +36,23 @@ jobs:
|
||||
# https://developer.nvidia.com/cuda-toolkit-archive
|
||||
cuda-version: ["11.8.0", "12.8.1"]
|
||||
steps:
|
||||
- name: Install tools
|
||||
run: |
|
||||
sudo apt-get update && sudo apt-get install -y \
|
||||
curl \
|
||||
ca-certificates \
|
||||
sudo \
|
||||
software-properties-common \
|
||||
wget \
|
||||
unzip \
|
||||
zip \
|
||||
git \
|
||||
build-essential \
|
||||
gcc \
|
||||
g++ \
|
||||
clang \
|
||||
ninja-build
|
||||
|
||||
- uses: actions/checkout@v4
|
||||
- uses: actions/setup-python@v5
|
||||
with:
|
||||
|
||||
@@ -11,7 +11,8 @@ ARG TARGET_ARCH="x64"
|
||||
ENV DEBIAN_FRONTEND=noninteractive \
|
||||
AGENT_TOOLSDIRECTORY=/opt/hostedtoolcache
|
||||
|
||||
RUN mkdir -p /opt/hostedtoolcache
|
||||
RUN mkdir -p /opt/hostedtoolcache \
|
||||
&& chown -R ubuntu:ubuntu /opt/hostedtoolcache
|
||||
|
||||
RUN apt-get update && apt-get install -y --no-install-recommends\
|
||||
curl \
|
||||
|
||||
Reference in New Issue
Block a user