From 565b3a4587403c0dc32be631151e296f43f5d269 Mon Sep 17 00:00:00 2001 From: Guillaume LEGENDRE Date: Thu, 26 Sep 2024 18:31:01 +0200 Subject: [PATCH 1/2] change runner cluster for A10 --- .github/workflows/fa2_unit_tests.yaml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/fa2_unit_tests.yaml b/.github/workflows/fa2_unit_tests.yaml index 342be45e..1f5c1aa5 100644 --- a/.github/workflows/fa2_unit_tests.yaml +++ b/.github/workflows/fa2_unit_tests.yaml @@ -19,7 +19,8 @@ on: jobs: tests: - runs-on: [single-gpu, nvidia-gpu, a10, ci] + runs-on: + group: aws-g5-4xlarge-plus container: image: runpod/pytorch:2.1.1-py3.10-cuda12.1.1-devel-ubuntu22.04 ports: From 676351b51910a3a60b39366eb1cccd463a692248 Mon Sep 17 00:00:00 2001 From: Guillaume LEGENDRE Date: Thu, 26 Sep 2024 18:32:22 +0200 Subject: [PATCH 2/2] change runner cluster for 8-t4 --- .github/workflows/3d_parallelism_unit_tests.yaml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/3d_parallelism_unit_tests.yaml b/.github/workflows/3d_parallelism_unit_tests.yaml index 28b70dd9..96dc9237 100644 --- a/.github/workflows/3d_parallelism_unit_tests.yaml +++ b/.github/workflows/3d_parallelism_unit_tests.yaml @@ -18,7 +18,8 @@ on: jobs: tests: - runs-on: [multi-gpu, nvidia-gpu, 8-t4, ci] + runs-on: + group: aws-g4dn-metal container: image: runpod/pytorch:2.1.1-py3.10-cuda12.1.1-devel-ubuntu22.04 ports: