From c121d3e48754b3bce100636f8d666deec58a44b7 Mon Sep 17 00:00:00 2001
From: mertalev <101130780+mertalev@users.noreply.github.com>
Date: Wed, 5 Mar 2025 20:49:58 -0500
Subject: [PATCH] set migraphx_home

---
 machine-learning/Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/machine-learning/Dockerfile b/machine-learning/Dockerfile
index 31553974f3..a7b67c60e4 100644
--- a/machine-learning/Dockerfile
+++ b/machine-learning/Dockerfile
@@ -40,7 +40,7 @@ RUN git apply /tmp/0001-disable-rocm-conv-algo-caching.patch
 RUN /bin/sh ./dockerfiles/scripts/install_common_deps.sh
 # Note: the `parallel` setting uses a substantial amount of RAM
 RUN ./build.sh --allow_running_as_root --config Release --build_wheel --update --build --parallel 17 --cmake_extra_defines\
-    ONNXRUNTIME_VERSION=1.20.1 --use_rocm --rocm_home=/opt/rocm --use_migraphx
+    ONNXRUNTIME_VERSION=1.20.1 --use_migraphx --migraphx_home /opt/rocm
 RUN mv /code/onnxruntime/build/Linux/Release/dist/*.whl /opt/
 
 FROM builder-${DEVICE} AS builder