test

huggingface · Aug 30, 2024 · 4d182c3 · 4d182c3
1 parent 0ad52b3
commit 4d182c3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/test_api_rocm.yaml b/.github/workflows/test_api_rocm.yaml
@@ -30,7 +30,8 @@ jobs:
     # the script in scripts/rocm_docker_devices.sh sets the environment variable ROCM_DOCKER_DEVICES
     # which is a string of the form --device /dev/kfd --device /dev/dri/renderD128 --device /dev/dri/renderD129 ...
     # that's used in the next job to mount the devices in the docker container
-    runs-on: ubuntu-latest
+    runs-on: [self-hosted, amd-gpu, single-gpu, mi250]
+
     steps:
       - name: Checkout
         uses: actions/checkout@v4