chore(ci): migrate runner configuration in GitHub workflows (#1978)

Signed-off-by: Adrien <[email protected]>

.github/workflows/doctests.yml

@@ -9,7 +9,8 @@ on:
 jobs:
   do-the-job:
     name: Start self-hosted EC2 runner
-    runs-on: [single-gpu, nvidia-gpu, t4, ci]
+    runs-on:
+      group: aws-g6-4xlarge-plus
     env:
       AWS_REGION: us-east-1
     steps:

.github/workflows/test_bettertransformer_gpu.yml

@@ -8,7 +8,8 @@ on:
 jobs:
   do-the-job:
     name: Start self-hosted EC2 runner
-    runs-on: [single-gpu, nvidia-gpu, t4, ci]
+    runs-on:
+      group: aws-g6-4xlarge-plus
     env:
       AWS_REGION: us-east-1
     steps:

.github/workflows/test_exporters_gpu.yml

@@ -15,7 +15,8 @@ jobs:
   do-the-job:
     if: ${{ (github.event_name == 'workflow_dispatch') || (github.event_name == 'schedule') || contains( github.event.pull_request.labels.*.name, 'gpu-test') }}
     name: Start self-hosted EC2 runner
-    runs-on: [single-gpu, nvidia-gpu, t4, ci]
+    runs-on:
+      group: aws-g6-4xlarge-plus
     env:
       AWS_REGION: us-east-1
     steps:

.github/workflows/test_gptq.yml

@@ -20,7 +20,8 @@ on:
 
 jobs:
   test_gptq:
-    runs-on: [single-gpu, nvidia-gpu, t4, ci]
+    runs-on:
+      group: aws-g6-4xlarge-plus
 
     steps:
       - name: Checkout code

.github/workflows/test_onnxruntime_gpu.yml

@@ -15,7 +15,8 @@ jobs:
   do-the-job:
     if: ${{ (github.event_name == 'workflow_dispatch') || (github.event_name == 'schedule') || contains( github.event.pull_request.labels.*.name, 'gpu-test') }}
     name: Start self-hosted EC2 runner
-    runs-on: [single-gpu, nvidia-gpu, t4, ci]
+    runs-on:
+      group: aws-g6-4xlarge-plus
     env:
       AWS_REGION: us-east-1
     steps:

.github/workflows/test_onnxruntime_train.yml

@@ -11,7 +11,8 @@ jobs:
   do-the-job:
     if: ${{ (github.event_name == 'workflow_dispatch') || (github.event_name == 'schedule') ||  contains( github.event.pull_request.labels.*.name, 'training')}}
     name: Run ORTTrainer test
-    runs-on: [single-gpu, nvidia-gpu, t4, ci]
+    runs-on:
+      group: aws-g6-4xlarge-plus
     env:
       AWS_REGION: us-east-1
     steps:
@@ -22,4 +23,4 @@ jobs:
           docker build -f tests/onnxruntime/docker/Dockerfile_onnxruntime_trainer -t onnxruntime/train .
       - name: Run test within docker container
         run: |
-          docker run --rm --gpus all -v /mnt/cache/.cache/huggingface:/root/.cache/huggingface --workdir=/workspace/optimum/tests onnxruntime/train:latest
+          docker run --rm --gpus all -v /mnt/cache/.cache/huggingface:/root/.cache/huggingface --workdir=/workspace/optimum/tests onnxruntime/train:latest