From dcf1e0c3b07f03724baabf0e37602a255ecc83c0 Mon Sep 17 00:00:00 2001
From: Changming Sun <chasun@microsoft.com>
Date: Wed, 9 Oct 2024 20:21:27 -0700
Subject: [PATCH] Re-enable CUDA 12 python package test pipeline (#22370)

### Description
It runs after "Python-CUDA-Packaging-Pipeline" that runs on a CPU
machine that skipped all tests.
This testing pipeline is for doing the tests.
---
 .../azure-pipelines/py-cuda-package-test-pipeline.yml | 11 ++++++-----
 .../templates/py-packaging-linux-test-cuda.yml        |  2 +-
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/tools/ci_build/github/azure-pipelines/py-cuda-package-test-pipeline.yml b/tools/ci_build/github/azure-pipelines/py-cuda-package-test-pipeline.yml
index 79b69bf34c..062b88a326 100644
--- a/tools/ci_build/github/azure-pipelines/py-cuda-package-test-pipeline.yml
+++ b/tools/ci_build/github/azure-pipelines/py-cuda-package-test-pipeline.yml
@@ -12,12 +12,13 @@ stages:
   - stage: Linux_Test_GPU_x86_64_stage
     dependsOn:
     jobs:
-      - template: stages/jobs/py-linux-cuda-package-test-job.yml
+      - template: templates/py-packaging-linux-test-cuda.yml
         parameters:
-          CudaVersion: '12.2'
+          arch: 'x86_64'
           machine_pool: 'Onnxruntime-Linux-GPU'
+          python_wheel_suffix: '_gpu'
           timeout: 480
-          build_id: ${{ parameters.build_id }}
-          project: ${{ parameters.project }}
-          pipeline: ${{ parameters.pipeline }}
+          docker_base_image: onnxruntimebuildcache.azurecr.io/internal/azureml/onnxruntime/build/cuda12_x64_ubi8_gcc12:20241008.1
+          trt_version: '10.4.0.26-1.cuda12.6'
+          cuda_version: '12.2'
 
diff --git a/tools/ci_build/github/azure-pipelines/templates/py-packaging-linux-test-cuda.yml b/tools/ci_build/github/azure-pipelines/templates/py-packaging-linux-test-cuda.yml
index 0c3cd60a71..4ca462bf96 100644
--- a/tools/ci_build/github/azure-pipelines/templates/py-packaging-linux-test-cuda.yml
+++ b/tools/ci_build/github/azure-pipelines/templates/py-packaging-linux-test-cuda.yml
@@ -100,7 +100,7 @@ jobs:
     inputs:
       targetType: filePath
       filePath: tools/ci_build/github/linux/run_python_dockertest.sh
-      arguments: -d GPU -c ${{parameters.cmake_build_type}} -i onnxruntimecuda118xtrt86build${{ parameters.arch }}
+      arguments: -d GPU -c ${{parameters.cmake_build_type}} -i onnxruntimecuda${{ replace(parameters.cuda_version, '.', '') }}xtrt86build${{ parameters.arch }}
 
   - task: mspremier.PostBuildCleanup.PostBuildCleanup-task.PostBuildCleanup@3
     displayName: 'Clean Agent Directories'