migraphx ci uses rocm docker file

microsoft · Jan 30, 2025 · fd30f52 · fd30f52
1 parent da42796
commit fd30f52
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 92 deletions.
diff --git a/tools/ci_build/github/azure-pipelines/linux-migraphx-ci-pipeline.yml b/tools/ci_build/github/azure-pipelines/linux-migraphx-ci-pipeline.yml
@@ -37,7 +37,7 @@ variables:
   - name: render
     value: 109
   - name: RocmVersion
-    value: 6.2.3
+    value: 6.3.2
 
 jobs:
 - job: Linux_Build
@@ -59,10 +59,10 @@ jobs:
 
   - template: templates/get-docker-image-steps.yml
     parameters:
-      Dockerfile: tools/ci_build/github/linux/docker/migraphx-ci-pipeline-env.Dockerfile
+      Dockerfile: tools/ci_build/github/linux/docker/Dockerfile.ubuntu_rocm
       Context: tools/ci_build/github/linux/docker
-      DockerBuildArgs: "--build-arg ROCM_VERSION=$(RocmVersion)"
-      Repository: onnxruntimetrainingmigraphx-cibuild-rocm$(RocmVersion)
+      DockerBuildArgs: "--build-arg ROCM_VERSION=$(RocmVersion) --build-arg USE_MIGRAPHX=1"
+      Repository: onnxruntime-migraphx-cibuild-rocm$(RocmVersion)
 
   - task: Cache@2
     inputs:
@@ -90,9 +90,10 @@ jobs:
           --volume $(CCACHE_DIR):/cache \
           -e CCACHE_DIR=/cache \
           --workdir /onnxruntime_src \
-          onnxruntimetrainingmigraphx-cibuild-rocm$(RocmVersion) \
+          onnxruntime-migraphx-cibuild-rocm$(RocmVersion) \
           /bin/bash -c "
             set -ex; \
+            source /ort/env/bin/activate; \
             env; \
             ccache -s; \
             python tools/ci_build/build.py \
@@ -104,6 +105,7 @@ jobs:
                 onnxruntime_USE_COMPOSABLE_KERNEL=OFF \
               --mpi_home /opt/ompi \
               --use_migraphx \
+              --migraphx_home /opt/rocm \
               --rocm_version=$(RocmVersion) \
               --rocm_home /opt/rocm \
               --nccl_home /opt/rocm \
@@ -158,10 +160,10 @@ jobs:
 
   - template: templates/get-docker-image-steps.yml
     parameters:
-      Dockerfile: tools/ci_build/github/linux/docker/migraphx-ci-pipeline-env.Dockerfile
+      Dockerfile: tools/ci_build/github/linux/docker/Dockerfile.ubuntu_rocm
       Context: tools/ci_build/github/linux/docker
-      DockerBuildArgs: "--build-arg ROCM_VERSION=$(RocmVersion)"
-      Repository: onnxruntimetrainingmigraphx-cibuild-rocm$(RocmVersion)
+      DockerBuildArgs: "--build-arg ROCM_VERSION=$(RocmVersion) --build-arg USE_MIGRAPHX=1"
+      Repository: onnxruntime-migraphx-cibuild-rocm$(RocmVersion)
 
   - task: CmdLine@2
     inputs:
@@ -177,9 +179,10 @@ jobs:
           --volume $(Build.SourcesDirectory):/onnxruntime_src \
           --volume $(Build.BinariesDirectory):/build \
           --workdir /build/Release \
-          onnxruntimetrainingmigraphx-cibuild-rocm$(RocmVersion) \
+          onnxruntime-migraphx-cibuild-rocm$(RocmVersion) \
           /bin/bash -c "
             set -ex; \
+            source /ort/env/bin/activate; \
             cd /build/Release && xargs -a /build/Release/perms.txt chmod a+x; \
             bash /onnxruntime_src/tools/ci_build/github/pai/pai_test_launcher.sh"
       workingDirectory: $(Build.SourcesDirectory)

diff --git a/tools/ci_build/github/linux/docker/migraphx-ci-pipeline-env.Dockerfile b/tools/ci_build/github/linux/docker/migraphx-ci-pipeline-env.Dockerfile