From 39eeb34d7718072f20bc5554ecbd56f3d470656b Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Thu, 20 Feb 2025 18:48:03 +0100
Subject: [PATCH] Working around the github runner thing.

---
 .github/workflows/build.yaml | 15 +++++++--------
 1 file changed, 7 insertions(+), 8 deletions(-)

diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index 3a2c7cd2c..2213f334d 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -262,14 +262,13 @@ jobs:
       cancel-in-progress: true
     runs-on:
       group: aws-g6-12xl-plus-priv-cache
-    container:
-      image: ${{ needs.build-and-push.outputs.docker_image }}
-      credentials:
-        username: ${{ secrets.DOCKERHUB_USERNAME }}
-        password: ${{ secrets.DOCKERHUB_PASSWORD }}
-      options: --gpus all --shm-size=8g
-
     steps:
+      - name: Initialize Docker Buildx
+        uses: docker/setup-buildx-action@v3
+        with:
+          install: true
+          buildkitd-config: /tmp/buildkitd.toml
       - name: Run C++/CUDA tests
         if: ${{ env.LABEL == 'ci-runtime' }}
-        run: /usr/local/tgi/bin/tgi_trtllm_backend_tests
+        run: docker run  --gpus all --shm-size=8g --entrypoint /usr/local/tgi/bin/tgi_trtllm_backend_tests ${{ needs.build-and-push.outputs.docker_image }}
+