diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml index 71cbd12b9..47408bc65 100644 --- a/.github/workflows/pr-test.yml +++ b/.github/workflows/pr-test.yml @@ -17,6 +17,8 @@ jobs: runner: linux.24xlarge - config: cuda runner: linux.aws.a100 + - config: cuda + runner: linux.dgx.b200 name: Test ${{ matrix.config }} env: DOCKER_IMAGE: 'ghcr.io/pytorch/torchbench:latest'