diff --git a/.github/workflows/fbgemm_gpu_ci_rocm.yml b/.github/workflows/fbgemm_gpu_ci_rocm.yml index f3fca6f5bf..4d04ccbab7 100644 --- a/.github/workflows/fbgemm_gpu_ci_rocm.yml +++ b/.github/workflows/fbgemm_gpu_ci_rocm.yml @@ -60,7 +60,7 @@ jobs: fail-fast: false matrix: host-machine: [ - { arch: x86, instance: "linux.24xlarge" }, + { arch: x86, instance: "gfx90a" }, ] container-image: [ "ubuntu:20.04" ] python-version: [ "3.8", "3.9", "3.10", "3.11", "3.12" ] @@ -138,7 +138,7 @@ jobs: fail-fast: false matrix: host-machine: [ - { arch: x86, instance: "rocm" }, + { arch: x86, instance: "gfx90a" }, ] # ROCm machines are limited, so we only test a subset of Python versions python-version: [ "3.12" ]