use linux runners

This commit is contained in:
Michael Yang 2024-02-06 13:35:54 -08:00
parent 59ec837ef6
commit 92b1a21f79

View file

@ -34,7 +34,7 @@ jobs:
matrix: matrix:
cuda-version: cuda-version:
- '11.8.0' - '11.8.0'
runs-on: ubuntu-latest runs-on: linux
container: nvidia/cuda:${{ matrix.cuda-version }}-devel-ubuntu20.04 container: nvidia/cuda:${{ matrix.cuda-version }}-devel-ubuntu20.04
steps: steps:
- run: | - run: |
@ -66,7 +66,7 @@ jobs:
rocm-version: rocm-version:
- '5.7.1' - '5.7.1'
- '6.0' - '6.0'
runs-on: ubuntu-latest runs-on: linux
container: rocm/dev-ubuntu-20.04:${{ matrix.rocm-version }} container: rocm/dev-ubuntu-20.04:${{ matrix.rocm-version }}
steps: steps:
- run: | - run: |