From fe22100549605bb642e0d70d959a293336147a6c Mon Sep 17 00:00:00 2001
From: Oliver Koenig <okoenig@nvidia.com>
Date: Thu, 20 Feb 2025 03:04:35 -0800
Subject: [PATCH] ADLR/megatron-lm!2721 - ci: Use larger runner for building
 the wheel

---
 .gitlab/stages/01.test.yml | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/.gitlab/stages/01.test.yml b/.gitlab/stages/01.test.yml
index 02fd7cb1fc..2bd532e3f7 100644
--- a/.gitlab/stages/01.test.yml
+++ b/.gitlab/stages/01.test.yml
@@ -356,12 +356,16 @@ test:pypi_build_wheel:
   image:
     name: quay.io/pypa/manylinux_2_28_x86_64
     entrypoint: ['']
+  services:
+    - name: docker:24.0.5-dind
+      variables:
+        HEALTHCHECK_TCP_PORT: '2376'
   tags: 
     - arch/amd64
     - env/prod
     - origin/jet-fleet
     - owner/jet-core
-    - purpose/utility
+    - purpose/builder-small
     - team/megatron
   variables:
     PUBLISH_DRYRUN: 'yes'