diff --git a/.github/workflows/h100-distributed.yml b/.github/workflows/h100-distributed.yml index a0a7495483d4..be19b8f961f4 100644 --- a/.github/workflows/h100-distributed.yml +++ b/.github/workflows/h100-distributed.yml @@ -37,7 +37,7 @@ jobs: needs: get-label-type with: runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" - runner: "linux.12xlarge" + runner: "linux.c7i.12xlarge" build-environment: linux-jammy-cuda12.8-py3.10-gcc11-sm90-dist docker-image-name: ci-image:pytorch-linux-jammy-cuda12.8-cudnn9-py3-gcc11 cuda-arch-list: '9.0' diff --git a/.github/workflows/xpu.yml b/.github/workflows/xpu.yml index 36ba62349f28..c6bdb06812e7 100644 --- a/.github/workflows/xpu.yml +++ b/.github/workflows/xpu.yml @@ -35,7 +35,7 @@ jobs: runner_prefix: ${{ needs.get-label-type.outputs.label-type }} build-environment: linux-jammy-xpu-n-1-py3.10 docker-image-name: ci-image:pytorch-linux-jammy-xpu-n-1-py3 - runner: linux.12xlarge + runner: linux.c7i.12xlarge test-matrix: | { include: [ { config: "default", shard: 1, num_shards: 6, runner: "linux.idc.xpu" }, @@ -56,7 +56,7 @@ jobs: runner_prefix: ${{ needs.get-label-type.outputs.label-type }} build-environment: linux-jammy-xpu-n-py3.10 docker-image-name: ci-image:pytorch-linux-jammy-xpu-n-py3 - runner: linux.12xlarge + runner: linux.c7i.12xlarge test-matrix: | { include: [ { config: "default", shard: 1, num_shards: 8, runner: "linux.idc.xpu" },