From cc6ef147583f8b583062e01e2288de4a7b4a46f3 Mon Sep 17 00:00:00 2001 From: Adrien Date: Wed, 31 Jul 2024 16:52:06 +0200 Subject: [PATCH] chore(ci): migrate runner configuration in GitHub workflows 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 d2879157b1..6269c58977 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 78a2995a11..99f4691084 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 e51d49a7c9..6d0d395a6c 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 7e7d3959a6..9b04589c67 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 b73369ec8b..e2337de710 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 7967f1b036..09a3a2090b 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