[ROCm] Add ROCm5.6 to python package pipeline (#16572)

Add ROCm5.6 to python package pipeline.
amd · Jul 4, 2023 · d540c7d · d540c7d
1 parent 13cc619
commit d540c7d
Showing 1 changed file with 15 additions and 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
@@ -38,15 +38,27 @@ stages:
   - template: templates/rocm.yml
     parameters:
       PythonVersion: '3.8'
-      RocmVersion: '5.5'
+      RocmVersion: '5.6'
+  - template: templates/rocm.yml
+    parameters:
+      PythonVersion: '3.9'
+      RocmVersion: '5.6'
+  - template: templates/rocm.yml
+    parameters:
+      PythonVersion: '3.10'
+      RocmVersion: '5.6'
+  - template: templates/rocm.yml
+    parameters:
+      PythonVersion: '3.8'
+      RocmVersion: '5.6'
       BuildConfig: 'RelWithDebInfo'
   - template: templates/rocm.yml
     parameters:
       PythonVersion: '3.9'
-      RocmVersion: '5.5'
+      RocmVersion: '5.6'
       BuildConfig: 'RelWithDebInfo'
   - template: templates/rocm.yml
     parameters:
       PythonVersion: '3.10'
-      RocmVersion: '5.5'
+      RocmVersion: '5.6'
       BuildConfig: 'RelWithDebInfo'