From b7e848e7f5d661665b7490b02e070fd4ba4363c1 Mon Sep 17 00:00:00 2001
From: Morgan Funtowicz <funtowiczmo@gmail.com>
Date: Tue, 7 Jan 2025 11:51:38 +0100
Subject: [PATCH] misc(ci): make the wf callable for reuse (bis)

---
 .github/workflows/trtllm_tests.yaml | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/.github/workflows/trtllm_tests.yaml b/.github/workflows/trtllm_tests.yaml
index a49695ad..ae741011 100644
--- a/.github/workflows/trtllm_tests.yaml
+++ b/.github/workflows/trtllm_tests.yaml
@@ -30,13 +30,17 @@ jobs:
 
   run-tests:
     needs: build-and-push
+    concurrency:
+      group: ${{ github.workflow }}-${{ github.job }}-trtllm-${{ github.head_ref || github.run_id }}
+      cancel-in-progress: true
     runs-on:
-      container:
-        image: ${{ needs.build-and-push.outputs.docker_image }}
-        credentials:
-          username: ${{ secrets.REGISTRY_USERNAME }}
-          password: ${{ secrets.REGISTRY_PASSWORD }}
-        options: --gpus all
+      group: aws-highmemory-64-plus-priv
+    container:
+      image: ${{ needs.build-and-push.outputs.docker_image }}
+      credentials:
+        username: ${{ secrets.REGISTRY_USERNAME }}
+        password: ${{ secrets.REGISTRY_PASSWORD }}
+      options: --gpus all --shm-size=8g
 
     steps:
       - name: Run C++/CUDA tests