From b65527f72be87b0ff0af9a68af0db929a84465df Mon Sep 17 00:00:00 2001
From: Changming Sun <chasun@microsoft.com>
Date: Sat, 4 Jan 2025 04:35:17 +0000
Subject: [PATCH] update

---
 .../py-cuda-alt-package-test-pipeline.yml     | 58 +++++++++----------
 .../py-cuda-package-test-pipeline.yml         | 34 +++++++++++
 2 files changed, 63 insertions(+), 29 deletions(-)

diff --git a/tools/ci_build/github/azure-pipelines/py-cuda-alt-package-test-pipeline.yml b/tools/ci_build/github/azure-pipelines/py-cuda-alt-package-test-pipeline.yml
index 27cb61942ff3a..8f26d1ae4976a 100644
--- a/tools/ci_build/github/azure-pipelines/py-cuda-alt-package-test-pipeline.yml
+++ b/tools/ci_build/github/azure-pipelines/py-cuda-alt-package-test-pipeline.yml
@@ -25,34 +25,34 @@ stages:
   - stage: Linux_Test_CUDA_Alt_x86_64_stage
     dependsOn:
     jobs:
-  - job: Python_Publishing_GPU
-    pool: 'onnxruntime-Ubuntu2204-AMD-CPU'
-    steps:
-    - checkout: none
-    - download: build
-      displayName: 'Download Pipeline Artifact - onnxruntime_gpu'
-      artifact: 'onnxruntime_gpu'
-      patterns: '*.whl'
-    - download: build
-      displayName: 'Download Pipeline Artifact - Win GPU 3.10'
-      artifact: 'win_gpu_wheel_3.10'
-      patterns: '*.whl'
-    - download: build
-      displayName: 'Download Pipeline Artifact - Win GPU 3.11'
-      artifact: 'win_gpu_wheel_3.11'
-      patterns: '*.whl'
-    - download: build
-      displayName: 'Download Pipeline Artifact - Win GPU 3.12'
-      artifact: 'win_gpu_wheel_3.12'
-      patterns: '*.whl'
-    - download: build
-      displayName: 'Download Pipeline Artifact - Win GPU 3.13'
-      artifact: 'win_gpu_wheel_3.13'
-      patterns: '*.whl'
+    - job: Python_Publishing_GPU
+      pool: 'onnxruntime-Ubuntu2204-AMD-CPU'
+      steps:
+      - checkout: none
+      - download: build
+        displayName: 'Download Pipeline Artifact - onnxruntime_gpu'
+        artifact: 'onnxruntime_gpu'
+        patterns: '*.whl'
+      - download: build
+        displayName: 'Download Pipeline Artifact - Win GPU 3.10'
+        artifact: 'win_gpu_wheel_3.10'
+        patterns: '*.whl'
+      - download: build
+        displayName: 'Download Pipeline Artifact - Win GPU 3.11'
+        artifact: 'win_gpu_wheel_3.11'
+        patterns: '*.whl'
+      - download: build
+        displayName: 'Download Pipeline Artifact - Win GPU 3.12'
+        artifact: 'win_gpu_wheel_3.12'
+        patterns: '*.whl'
+      - download: build
+        displayName: 'Download Pipeline Artifact - Win GPU 3.13'
+        artifact: 'win_gpu_wheel_3.13'
+        patterns: '*.whl'
 
-    - script: find $(Pipeline.Workspace) -name \*win_amd64.whl -exec mv {} $(Pipeline.Workspace)/build/onnxruntime_gpu \;
-      displayName: 'Merge files together'
+      - script: find $(Pipeline.Workspace) -name \*win_amd64.whl -exec mv {} $(Pipeline.Workspace)/build/onnxruntime_gpu \;
+        displayName: 'Merge files together'
 
-    - publish: $(Pipeline.Workspace)/build/onnxruntime_gpu
-      artifact: whl
-      displayName: Republish artifacts
\ No newline at end of file
+      - publish: $(Pipeline.Workspace)/build/onnxruntime_gpu
+        artifact: whl
+        displayName: Republish artifacts
\ No newline at end of file
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 307415b7be16f..b0a40cf31912a 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
@@ -21,3 +21,37 @@ stages:
           docker_base_image: onnxruntimebuildcache.azurecr.io/internal/azureml/onnxruntime/build/cuda12_x64_ubi8_gcc12:20241120.3
           cuda_version: '12.2'
 
+  - stage: Linux_Test_CUDA_Alt_x86_64_stage
+    dependsOn:
+    jobs:
+    - job: Python_Publishing_GPU
+      pool: 'onnxruntime-Ubuntu2204-AMD-CPU'
+      steps:
+      - checkout: none
+      - download: build
+        displayName: 'Download Pipeline Artifact - onnxruntime_gpu'
+        artifact: 'onnxruntime_gpu'
+        patterns: '*.whl'
+      - download: build
+        displayName: 'Download Pipeline Artifact - Win GPU 3.10'
+        artifact: 'win_gpu_wheel_3.10'
+        patterns: '*.whl'
+      - download: build
+        displayName: 'Download Pipeline Artifact - Win GPU 3.11'
+        artifact: 'win_gpu_wheel_3.11'
+        patterns: '*.whl'
+      - download: build
+        displayName: 'Download Pipeline Artifact - Win GPU 3.12'
+        artifact: 'win_gpu_wheel_3.12'
+        patterns: '*.whl'
+      - download: build
+        displayName: 'Download Pipeline Artifact - Win GPU 3.13'
+        artifact: 'win_gpu_wheel_3.13'
+        patterns: '*.whl'
+
+      - script: find $(Pipeline.Workspace) -name \*win_amd64.whl -exec mv {} $(Pipeline.Workspace)/build/onnxruntime_gpu \;
+        displayName: 'Merge files together'
+
+      - publish: $(Pipeline.Workspace)/build/onnxruntime_gpu
+        artifact: whl
+        displayName: Republish artifacts
\ No newline at end of file