From d3893bc8ad36ca75f980069692faf7f34e2a611a Mon Sep 17 00:00:00 2001 From: Morgan Funtowicz Date: Fri, 10 Jan 2025 15:11:44 +0100 Subject: [PATCH] misc(ci): wtf3 --- .github/workflows/build_trtllm.yaml | 2 +- .github/workflows/trtllm_tests.yaml | 4 ++++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/.github/workflows/build_trtllm.yaml b/.github/workflows/build_trtllm.yaml index c71d1628..f6badc3b 100644 --- a/.github/workflows/build_trtllm.yaml +++ b/.github/workflows/build_trtllm.yaml @@ -25,7 +25,7 @@ jobs: group: ${{ github.workflow }}-${{ github.job }}-build-and-push-${{ github.head_ref || github.run_id }} outputs: docker_image: ${{ steps.final.outputs.docker_image }} - label: ${ steps.final.outputs.label }} + label: ${{ steps.final.outputs.label }} runs-on: group: ${{ inputs.runs-on }} steps: diff --git a/.github/workflows/trtllm_tests.yaml b/.github/workflows/trtllm_tests.yaml index 2665d98c..9c9048f6 100644 --- a/.github/workflows/trtllm_tests.yaml +++ b/.github/workflows/trtllm_tests.yaml @@ -46,7 +46,11 @@ jobs: options: --gpus all --shm-size=8g steps: + - name: whoami + run: | + echo "Container: ${{ needs.build-backend.outputs.docker_image }}" - name: List binaries run: ls -alh /usr/local/tgi + - name: Run C++/CUDA tests run: /usr/local/tgi/bin/tgi_trtllm_backend_tests \ No newline at end of file