From 2a6d88d9925f64caf1d509d96253802d8f8e3365 Mon Sep 17 00:00:00 2001
From: Morgan Funtowicz <funtowiczmo@gmail.com>
Date: Mon, 6 Jan 2025 15:35:59 +0100
Subject: [PATCH] misc(ci): gogogo

---
 .github/workflows/build_trtllm.yaml | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/build_trtllm.yaml b/.github/workflows/build_trtllm.yaml
index ebfe068c..b53e9892 100644
--- a/.github/workflows/build_trtllm.yaml
+++ b/.github/workflows/build_trtllm.yaml
@@ -127,9 +127,12 @@ jobs:
         id: final
         run: |
           echo "docker_image=registry.internal.huggingface.tech/api-inference/community/text-generation-inference:sha-${{ env.GITHUB_SHA_SHORT }}${{ env.LABEL }}" >> "$GITHUB_OUTPUT"
-
+          echo "label=${{ env.LABEL }}" >> "$GITHUB_OUTPUT"
   tests:
     needs: build-and-push
+    concurrency:
+      group: ${{ github.workflow }}-${{ github.job }}-${{ needs.build-and-push.outputs.label }}-${{ github.head_ref || github.run_id }}
+      cancel-in-progress: true
     runs-on:
       group: aws-g6-12xl-plus-priv-cache
     env: