CI: Move compute-sanitzer setup into own step after CTK setup

Because the sanitizer commands depend on the version of the sanitizer
we need to be able to run the sanitzer to set the sanitizer cmd.
Thus, we need to setup the sanitzer after it is installed.

Author: carterbox

.github/workflows/test-wheel-linux.yml

@@ -74,19 +74,6 @@ jobs:
             fi
           fi
 
-          if [[ "${{ inputs.python-version }}" == "3.12" && "${{ inputs.local-ctk }}" == 1 ]]; then
-            # Use single quotes to delay expansion of $CUDA_HOME until after mini-CTK step is complete
-            COMPUTE_SANITIZER='${CUDA_HOME}/bin/compute-sanitizer'
-            COMPUTE_SANITIZER_VERSION=$(${COMPUTE_SANITIZER} --version | grep -Eo "[0-9]{4}\.[0-9]\.[0-9]" | sed -e 's/\.//g')
-            SANITIZER_CMD="${COMPUTE_SANITIZER} --target-processes=all --launch-timeout=0 --tool=memcheck --error-exitcode=1"
-            if [[ "$COMPUTE_SANITIZER_VERSION" -ge 202111 ]]; then
-                SANITIZER_CMD="${SANITIZER_CMD} --padding=32"
-            fi
-          else
-            COMPUTE_SANITIZER_VERSION="None"
-            SANITIZER_CMD=""
-          fi
-
           # make outputs from the previous job as env vars
           CUDA_CORE_ARTIFACT_BASENAME="cuda-core-python${PYTHON_VERSION_FORMATTED}-${{ inputs.host-platform }}"
           echo "PYTHON_VERSION_FORMATTED=${PYTHON_VERSION_FORMATTED}" >> $GITHUB_ENV
@@ -99,8 +86,6 @@ jobs:
           echo "CUDA_BINDINGS_ARTIFACTS_DIR=$(realpath "$REPO_DIR/cuda_bindings/dist")" >> $GITHUB_ENV
           echo "SKIP_CUDA_BINDINGS_TEST=${SKIP_CUDA_BINDINGS_TEST}" >> $GITHUB_ENV
           echo "SKIP_CUDA_CORE_CYTHON_TEST=${SKIP_CUDA_CORE_CYTHON_TEST}" >> $GITHUB_ENV
-          echo "COMPUTE_SANITIZER_VERSION=${COMPUTE_SANITIZER_VERSION}" >> $GITHUB_ENV
-          echo "SANITIZER_CMD=${SANITIZER_CMD}" >> $GITHUB_ENV
 
       - name: Install dependencies
         uses: ./.github/actions/install_unix_deps
@@ -195,6 +180,22 @@ jobs:
           host-platform: ${{ inputs.host-platform }}
           cuda-version: ${{ inputs.cuda-version }}
 
+      - name: Set up compute-saniziter
+        run: |
+          if [[ "${{ inputs.python-version }}" == "3.12" && "${{ inputs.local-ctk }}" == 1 ]]; then
+            COMPUTE_SANITIZER="${CUDA_HOME}/bin/compute-sanitizer"
+            COMPUTE_SANITIZER_VERSION=$(${COMPUTE_SANITIZER} --version | grep -Eo "[0-9]{4}\.[0-9]\.[0-9]" | sed -e 's/\.//g')
+            SANITIZER_CMD="${COMPUTE_SANITIZER} --target-processes=all --launch-timeout=0 --tool=memcheck --error-exitcode=1"
+            if [[ "$COMPUTE_SANITIZER_VERSION" -ge 202111 ]]; then
+                SANITIZER_CMD="${SANITIZER_CMD} --padding=32"
+            fi
+          else
+            COMPUTE_SANITIZER_VERSION="None"
+            SANITIZER_CMD=""
+          fi
+          echo "COMPUTE_SANITIZER_VERSION=${COMPUTE_SANITIZER_VERSION}" >> $GITHUB_ENV
+          echo "SANITIZER_CMD=${SANITIZER_CMD}" >> $GITHUB_ENV
+
       - name: Run cuda.bindings tests
         if: ${{ env.SKIP_CUDA_BINDINGS_TEST == '0' }}
         run: |