diff --git a/.github/workflows/publish-centml-pypi.yaml b/.github/workflows/publish-centml-pypi.yaml
index de9abae37..be32e50db 100644
--- a/.github/workflows/publish-centml-pypi.yaml
+++ b/.github/workflows/publish-centml-pypi.yaml
@@ -80,7 +80,8 @@ jobs:
     strategy:
       fail-fast: false
       matrix: ${{ fromJSON(needs.list-test-dirs.outputs.matrix) }}
-    runs-on: arc-runner-set
+    runs-on:
+      group: 'arc-l4'
     container:
       image: nvidia/cuda:12.4.0-devel-ubuntu20.04
     steps:
diff --git a/.github/workflows/publish-pypi.yaml b/.github/workflows/publish-pypi.yaml
index 099f4a96a..416b55bd9 100644
--- a/.github/workflows/publish-pypi.yaml
+++ b/.github/workflows/publish-pypi.yaml
@@ -56,7 +56,8 @@ jobs:
     strategy:
       fail-fast: false
       matrix: ${{ fromJSON(needs.list-test-dirs.outputs.matrix) }}
-    runs-on: arc-runner-set
+    runs-on:
+      group: 'arc-l4'
     container:
       image: nvidia/cuda:12.6.2-devel-ubuntu22.04
     steps:
diff --git a/.github/workflows/tests.yaml b/.github/workflows/tests.yaml
index 06741a8d1..5f92ab39b 100644
--- a/.github/workflows/tests.yaml
+++ b/.github/workflows/tests.yaml
@@ -12,9 +12,7 @@ on:
         type: choice
         options:
           - arc-l4
-          - arc-a10
-          - arc-h100
-        description: 'Runner group to run tests. (arc-l4, arc-a10, arc-h100)'
+        description: 'Runner group to run tests. (arc-l4)'
         required: false
         default: arc-l4
 
@@ -52,7 +50,7 @@ jobs:
 
   build-docs:
     runs-on:
-      arc-runner-set
+      group: ${{ inputs.runner_group || 'arc-l4' }}
     container:
       image: nvidia/cuda:12.6.2-devel-ubuntu22.04
       options: --gpus all
@@ -112,7 +110,7 @@ jobs:
       fail-fast: false
       matrix: ${{ fromJSON(needs.list-test-dirs.outputs.matrix) }}
     runs-on:
-      arc-runner-set
+      group: ${{ inputs.runner_group || 'arc-l4' }}
     container:
       image: nvidia/cuda:12.6.2-devel-ubuntu22.04
     steps: