Update runner labels and rocm version

ROCm · Sep 27, 2024 · 79351ec · 79351ec
1 parent 78a52d6
commit 79351ec
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/.github/workflows/fbgemm_gpu_ci_rocm.yml b/.github/workflows/fbgemm_gpu_ci_rocm.yml
@@ -61,11 +61,11 @@ jobs:
       fail-fast: false
       matrix:
         host-machine: [
-          { arch: x86, instance: "linux.24xlarge" },
+          { arch: x86, instance: "gfx90a" },
         ]
         container-image: [ "ubuntu:20.04" ]
         python-version: [ "3.9", "3.10", "3.11", "3.12" ]
-        rocm-version: [ "6.1" ]
+        rocm-version: [ "6.2" ]
         compiler: [ "gcc", "clang" ]
 
     steps:
@@ -141,11 +141,11 @@ 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" ]
-        rocm-version: [ "6.1" ]
+        rocm-version: [ "6.2" ]
         compiler: [ "gcc", "clang" ]
     needs: build_artifact