From d4fe4b5b514fee5dea3fa5150336883031417c4a Mon Sep 17 00:00:00 2001
From: Jian Chen <cjian@microsoft.com>
Date: Wed, 22 May 2024 13:36:02 -0700
Subject: [PATCH] Replace ubuntu-latest with onnxruntime-Ubuntu2204-AMD-CPU
 (#20736)

### Description
<!-- Describe your changes. -->



### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->
---
 .../azure-pipelines/c-api-noopenmp-packaging-pipelines.yml    | 4 ++--
 .../github/azure-pipelines/py-package-test-pipeline.yml       | 2 +-
 .../azure-pipelines/stages/set_packaging_variables_stage.yml  | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/tools/ci_build/github/azure-pipelines/c-api-noopenmp-packaging-pipelines.yml b/tools/ci_build/github/azure-pipelines/c-api-noopenmp-packaging-pipelines.yml
index cae34d451c..4408e3d244 100644
--- a/tools/ci_build/github/azure-pipelines/c-api-noopenmp-packaging-pipelines.yml
+++ b/tools/ci_build/github/azure-pipelines/c-api-noopenmp-packaging-pipelines.yml
@@ -127,7 +127,7 @@ stages:
   jobs:
   - job: D1
     pool:
-      vmImage: ubuntu-latest
+      name: 'onnxruntime-Ubuntu2204-AMD-CPU'
     variables:
       MyVar: $[stageDependencies.Setup.Set_Variables.outputs['Set_Release_Version_Suffix.ReleaseVersionSuffix']]
       BuildDate: $[stageDependencies.Setup.Set_Variables.outputs['Set_Build_Date.BuildDate']]
@@ -149,7 +149,7 @@ stages:
   jobs:
   - job: Download_Java_Tools
     pool:
-      vmImage: ubuntu-latest
+      name: 'onnxruntime-Ubuntu2204-AMD-CPU'
     steps:
     - checkout: none
     - task: mspremier.PostBuildCleanup.PostBuildCleanup-task.PostBuildCleanup@3
diff --git a/tools/ci_build/github/azure-pipelines/py-package-test-pipeline.yml b/tools/ci_build/github/azure-pipelines/py-package-test-pipeline.yml
index acec6f501e..59c3427f88 100644
--- a/tools/ci_build/github/azure-pipelines/py-package-test-pipeline.yml
+++ b/tools/ci_build/github/azure-pipelines/py-package-test-pipeline.yml
@@ -50,7 +50,7 @@ stages:
         job_name: Test_LINUX_x86_64_Wheels
         itemPattern: '*/*manylinux*x86_64.whl'
         machine_pool:
-          vmImage: 'ubuntu-22.04'
+          name: 'onnxruntime-Ubuntu2204-AMD-CPU'
 
 # ****The following Stage depend on all previous tags. ***
 
diff --git a/tools/ci_build/github/azure-pipelines/stages/set_packaging_variables_stage.yml b/tools/ci_build/github/azure-pipelines/stages/set_packaging_variables_stage.yml
index 616dc5495d..3e2b3b585d 100644
--- a/tools/ci_build/github/azure-pipelines/stages/set_packaging_variables_stage.yml
+++ b/tools/ci_build/github/azure-pipelines/stages/set_packaging_variables_stage.yml
@@ -7,7 +7,7 @@ stages:
   jobs:
   - job: Set_Variables
     pool:
-      vmImage: ubuntu-latest
+      name: 'onnxruntime-Ubuntu2204-AMD-CPU'
     steps:
     - checkout: none
     - task: mspremier.PostBuildCleanup.PostBuildCleanup-task.PostBuildCleanup@3