diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml index 0e3c7ece..051a0f82 100644 --- a/.github/workflows/test.yaml +++ b/.github/workflows/test.yaml @@ -34,7 +34,7 @@ jobs: matrix: cuda-version: - '11.8.0' - runs-on: ubuntu-latest + runs-on: linux container: nvidia/cuda:${{ matrix.cuda-version }}-devel-ubuntu20.04 steps: - run: | @@ -59,14 +59,12 @@ jobs: name: cuda-${{ matrix.cuda-version }}-libraries path: llm/llama.cpp/build/**/lib/* generate-rocm: - # TODO: re-enable once runners get more storage - if: ${{ false }} strategy: matrix: rocm-version: - '5.7.1' - '6.0' - runs-on: ubuntu-latest + runs-on: linux container: rocm/dev-ubuntu-20.04:${{ matrix.rocm-version }} steps: - run: |