mirror of
https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git
synced 2026-07-01 01:27:54 -04:00
update test workflow
This commit is contained in:
@@ -25,6 +25,8 @@ jobs:
|
|||||||
runs-on:
|
runs-on:
|
||||||
- self-hosted
|
- self-hosted
|
||||||
timeout-minutes: 1000
|
timeout-minutes: 1000
|
||||||
|
container:
|
||||||
|
image: ubuntu:22.04
|
||||||
strategy:
|
strategy:
|
||||||
fail-fast: false
|
fail-fast: false
|
||||||
matrix:
|
matrix:
|
||||||
@@ -34,6 +36,23 @@ jobs:
|
|||||||
# https://developer.nvidia.com/cuda-toolkit-archive
|
# https://developer.nvidia.com/cuda-toolkit-archive
|
||||||
cuda-version: ["11.8.0", "12.8.1"]
|
cuda-version: ["11.8.0", "12.8.1"]
|
||||||
steps:
|
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/checkout@v4
|
||||||
- uses: actions/setup-python@v5
|
- uses: actions/setup-python@v5
|
||||||
with:
|
with:
|
||||||
|
|||||||
@@ -11,7 +11,8 @@ ARG TARGET_ARCH="x64"
|
|||||||
ENV DEBIAN_FRONTEND=noninteractive \
|
ENV DEBIAN_FRONTEND=noninteractive \
|
||||||
AGENT_TOOLSDIRECTORY=/opt/hostedtoolcache
|
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\
|
RUN apt-get update && apt-get install -y --no-install-recommends\
|
||||||
curl \
|
curl \
|
||||||
|
|||||||
Reference in New Issue
Block a user