test mi210

huggingface · Oct 2, 2023 · a9bb027 · a9bb027
1 parent 45efab9
commit a9bb027
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/.github/workflows/test-amd.yaml b/.github/workflows/test-amd.yaml
@@ -19,7 +19,7 @@ env:
 jobs:
   check_runners:
     name: Check Runners 1 (multi-gpu) 
-    runs-on: [self-hosted, docker-gpu, amd-gpu, multi-gpu, mi250]
+    runs-on: [self-hosted, docker-gpu, amd-gpu, multi-gpu, mi210]
     container:
       image: huggingface/transformers-pytorch-amd-gpu-push-ci
       options: --device /dev/kfd --device /dev/dri --shm-size "16gb" --env HIP_VISIBLE_DEVICES --env ROCR_VISIBLE_DEVICES --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
@@ -33,7 +33,7 @@ jobs:
           echo "Output HIP: $ROCR_VISIBLE_DEVICES"
   check_runners2:
     name: Check Runners 2 (multi-gpu)   
-    runs-on: [self-hosted, docker-gpu, amd-gpu, multi-gpu, mi250]
+    runs-on: [self-hosted, docker-gpu, amd-gpu, multi-gpu, mi210]
     container:
       image: huggingface/transformers-pytorch-amd-gpu-push-ci
       options: --device /dev/kfd --device /dev/dri --shm-size "16gb" --env HIP_VISIBLE_DEVICES --env ROCR_VISIBLE_DEVICES --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
@@ -47,7 +47,7 @@ jobs:
           echo "Output ROCR: $ROCR_VISIBLE_DEVICES"
   check_runners3:
     name: Check Runners 1 (single-gpu)    
-    runs-on: [self-hosted, docker-gpu, amd-gpu, single-gpu, mi250]
+    runs-on: [self-hosted, docker-gpu, amd-gpu, single-gpu, mi210]
     container:
       image: huggingface/transformers-pytorch-amd-gpu-push-ci
       options: --device /dev/kfd --device /dev/dri --shm-size "16gb" --env HIP_VISIBLE_DEVICES --env ROCR_VISIBLE_DEVICES --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
@@ -62,7 +62,7 @@ jobs:
   
   check_runners4:
     name: Check Runners 2 (single-gpu)    
-    runs-on: [self-hosted, docker-gpu, amd-gpu, single-gpu, mi250]
+    runs-on: [self-hosted, docker-gpu, amd-gpu, single-gpu, mi210]
     container:
       image: huggingface/transformers-pytorch-amd-gpu-push-ci
       options: --device /dev/kfd --device /dev/dri --shm-size "16gb" --env HIP_VISIBLE_DEVICES --env ROCR_VISIBLE_DEVICES --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
@@ -77,7 +77,7 @@ jobs:
   
   check_runners5:
     name: Check Runners 3 (single-gpu)    
-    runs-on: [self-hosted, docker-gpu, amd-gpu, single-gpu, mi250]
+    runs-on: [self-hosted, docker-gpu, amd-gpu, single-gpu, mi210]
     container:
       image: huggingface/transformers-pytorch-amd-gpu-push-ci
       options: --device /dev/kfd --device /dev/dri --shm-size "16gb" --env HIP_VISIBLE_DEVICES --env ROCR_VISIBLE_DEVICES --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
@@ -92,7 +92,7 @@ jobs:
   
   check_runners6:
     name: Check Runners 4 (single-gpu)    
-    runs-on: [self-hosted, docker-gpu, amd-gpu, single-gpu, mi250]
+    runs-on: [self-hosted, docker-gpu, amd-gpu, single-gpu, mi210]
     container:
       image: huggingface/transformers-pytorch-amd-gpu-push-ci
       options: --device /dev/kfd --device /dev/dri --shm-size "16gb" --env HIP_VISIBLE_DEVICES --env ROCR_VISIBLE_DEVICES --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/