From 676351b51910a3a60b39366eb1cccd463a692248 Mon Sep 17 00:00:00 2001
From: Guillaume LEGENDRE <glegendre01@gmail.com>
Date: Thu, 26 Sep 2024 18:32:22 +0200
Subject: [PATCH] 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: