From b5fd2a6a80ab429c50dbf3078966aaccc79e7fdb Mon Sep 17 00:00:00 2001 From: PeixuanZuo <94887879+PeixuanZuo@users.noreply.github.com> Date: Thu, 22 Dec 2022 10:01:40 +0800 Subject: [PATCH] [ROCm] Add ROCm5.4 to python package pipeline (#14012) Add ROCm5.4 to python package pipeline. The download link of ROCm5.4 nightly build whl is https://download.onnxruntime.ai/onnxruntime_nightly_rocm54.html The download linkd of ROCm5.4 nightly build whl with profiling is https://download.onnxruntime.ai/onnxruntime_nightly_rocm54.profiling.html Co-authored-by: peixuanzuo --- .../orttraining-py-packaging-pipeline-rocm.yml | 18 +++++++++++++++--- 1 file changed, 15 insertions(+), 3 deletions(-) diff --git a/tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-rocm.yml b/tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-rocm.yml index df8ef8ce9f..aefa3f7962 100644 --- a/tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-rocm.yml +++ b/tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-rocm.yml @@ -38,15 +38,27 @@ stages: - template: templates/rocm.yml parameters: PythonVersion: '3.7' - RocmVersion: '5.3.2' + RocmVersion: '5.4' + - template: templates/rocm.yml + parameters: + PythonVersion: '3.8' + RocmVersion: '5.4' + - template: templates/rocm.yml + parameters: + PythonVersion: '3.9' + RocmVersion: '5.4' + - template: templates/rocm.yml + parameters: + PythonVersion: '3.7' + RocmVersion: '5.4' BuildConfig: 'RelWithDebInfo' - template: templates/rocm.yml parameters: PythonVersion: '3.8' - RocmVersion: '5.3.2' + RocmVersion: '5.4' BuildConfig: 'RelWithDebInfo' - template: templates/rocm.yml parameters: PythonVersion: '3.9' - RocmVersion: '5.3.2' + RocmVersion: '5.4' BuildConfig: 'RelWithDebInfo'