From c4abc206f4cb3f543c73979b70ff14ee6c075f9f Mon Sep 17 00:00:00 2001
From: Morgan Funtowicz <funtowiczmo@gmail.com>
Date: Wed, 8 Jan 2025 23:59:08 +0100
Subject: [PATCH] misc(ci):run the tests on GPU instance

---
 .github/workflows/trtllm_tests.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/trtllm_tests.yaml b/.github/workflows/trtllm_tests.yaml
index 090870ad..2b818f45 100644
--- a/.github/workflows/trtllm_tests.yaml
+++ b/.github/workflows/trtllm_tests.yaml
@@ -37,7 +37,7 @@ jobs:
       group: ${{ github.workflow }}-${{ github.job }}-trtllm-${{ github.head_ref || github.run_id }}
       cancel-in-progress: true
     runs-on:
-      group: aws-highmemory-64-plus-priv
+      group: aws-g6-12xl-plus-priv-cache
     container:
       image: ${{ needs.build-and-push.outputs.docker_image }}
       credentials: