Skip to content

ci: switch to amazon runners #20

ci: switch to amazon runners

ci: switch to amazon runners #20

Workflow file for this run

name: Unit Tests
on:
push:
branches:
- main
pull_request:
jobs:
unittest:
strategy:
matrix:
include:
- runs-on: "linux.2xlarge"
gpu-arch-type: "cpu"
gpu-arch-version: ""
- runs-on: "linux.4xlarge.nvidia.gpu"
gpu-arch-type: "cuda"
gpu-arch-version: "12.1"
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
with:
timeout: 120
runner: ${{ matrix.runs-on }}
gpu-arch-type: ${{ matrix.gpu-arch-type }}
gpu-arch-version: ${{ matrix.gpu-arch-version }}
script: |
conda create -n venv python=3.10 -y
conda activate venv
yum install -y protobuf-compiler
python -m pip install --upgrade pip
pip install -e .[dev] -v
pytest -v
cargo test -v