update

microsoft · Jan 4, 2025 · b65527f · b65527f
1 parent f19db1c
commit b65527f
Show file tree

Hide file tree

Showing 2 changed files with 63 additions and 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
@@ -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
+      - publish: $(Pipeline.Workspace)/build/onnxruntime_gpu
+        artifact: whl
+        displayName: Republish artifacts
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
@@ -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