From cfaece8dfb48ec5130c5ca8b7ed3f398141768b6 Mon Sep 17 00:00:00 2001 From: Adrien Date: Mon, 12 Aug 2024 12:40:21 +0200 Subject: [PATCH] chore(ci): migrate runner configuration in GitHub workflows (#1978) Signed-off-by: Adrien --- .github/workflows/doctests.yml | 3 ++- .github/workflows/test_bettertransformer_gpu.yml | 3 ++- .github/workflows/test_exporters_gpu.yml | 3 ++- .github/workflows/test_gptq.yml | 3 ++- .github/workflows/test_onnxruntime_gpu.yml | 3 ++- .github/workflows/test_onnxruntime_train.yml | 5 +++-- 6 files changed, 13 insertions(+), 7 deletions(-) diff --git a/.github/workflows/doctests.yml b/.github/workflows/doctests.yml index d2879157b16..6269c589774 100644 --- a/.github/workflows/doctests.yml +++ b/.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: diff --git a/.github/workflows/test_bettertransformer_gpu.yml b/.github/workflows/test_bettertransformer_gpu.yml index 78a2995a118..99f46910840 100644 --- a/.github/workflows/test_bettertransformer_gpu.yml +++ b/.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: diff --git a/.github/workflows/test_exporters_gpu.yml b/.github/workflows/test_exporters_gpu.yml index e51d49a7c9f..6d0d395a6c9 100644 --- a/.github/workflows/test_exporters_gpu.yml +++ b/.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: diff --git a/.github/workflows/test_gptq.yml b/.github/workflows/test_gptq.yml index 7e7d3959a6b..9b04589c670 100644 --- a/.github/workflows/test_gptq.yml +++ b/.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 diff --git a/.github/workflows/test_onnxruntime_gpu.yml b/.github/workflows/test_onnxruntime_gpu.yml index b73369ec8b2..e2337de7109 100644 --- a/.github/workflows/test_onnxruntime_gpu.yml +++ b/.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: diff --git a/.github/workflows/test_onnxruntime_train.yml b/.github/workflows/test_onnxruntime_train.yml index 7967f1b036b..09a3a2090bc 100644 --- a/.github/workflows/test_onnxruntime_train.yml +++ b/.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 \ No newline at end of file + docker run --rm --gpus all -v /mnt/cache/.cache/huggingface:/root/.cache/huggingface --workdir=/workspace/optimum/tests onnxruntime/train:latest