From fb487d0fedb9b91ef2d73024d407a9384aa657d3 Mon Sep 17 00:00:00 2001
From: Leo Fang <leof@nvidia.com>
Date: Sun, 8 Dec 2024 05:41:57 +0000
Subject: [PATCH] fix arg passing

---
 .github/actions/test/action.yml         | 4 ++--
 .github/workflows/gh-build-and-test.yml | 3 ++-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/.github/actions/test/action.yml b/.github/actions/test/action.yml
index 0a1e621a..c82a8450 100644
--- a/.github/actions/test/action.yml
+++ b/.github/actions/test/action.yml
@@ -46,8 +46,8 @@ runs:
     - name: Set up CTK cache variable
       shell: bash --noprofile --norc -xeuo pipefail {0}
       run: |
-        echo "CTK_CACHE_KEY=mini-ctk-${CTK_BUILD_VER}-${{ inputs.host-platform }}" >> $GITHUB_ENV
-        echo "CTK_CACHE_FILENAME=mini-ctk-${CTK_BUILD_VER}-${{ inputs.host-platform }}.tar.gz" >> $GITHUB_ENV
+        echo "CTK_CACHE_KEY=mini-ctk-${CTK_BUILD_VER}-${HOST_PLATFORM}" >> $GITHUB_ENV
+        echo "CTK_CACHE_FILENAME=mini-ctk-${CTK_BUILD_VER}-${HOST_PLATFORM}.tar.gz" >> $GITHUB_ENV
 
     - name: Download CTK cache
       id: ctk-get-cache
diff --git a/.github/workflows/gh-build-and-test.yml b/.github/workflows/gh-build-and-test.yml
index 556c47f2..06f6a168 100644
--- a/.github/workflows/gh-build-and-test.yml
+++ b/.github/workflows/gh-build-and-test.yml
@@ -75,7 +75,7 @@ jobs:
 
   test:
     # TODO: improve the name once a separate test matrix is defined
-    name: Test (CUDA ${{ inputs.cuda-version }}, Use container=${{ inputs.use-container }})
+    name: Test (CUDA ${{ inputs.cuda-version }})
     # TODO: enable testing once linux-aarch64 & win-64 GPU runners are up
     if: ${{ (github.repository_owner == 'nvidia') &&
              startsWith(inputs.host-platform, 'linux-x64') }}
@@ -121,3 +121,4 @@ jobs:
           CUDA_BINDINGS_ARTIFACTS_DIR: ${{ needs.build.outputs.CUDA_BINDINGS_ARTIFACTS_DIR }}
           PYTHON_VERSION: ${{ inputs.python-version }}
           CTK_BUILD_VER: ${{ inputs.cuda-version }}
+          HOST_PLATFORM: ${{ inputs.host-platform }}