From e484d1f35e1995abac90b01a4e3f3d2e21c87bae Mon Sep 17 00:00:00 2001
From: JeniferC99 <150404595+JeniferC99@users.noreply.github.com>
Date: Fri, 17 Oct 2025 14:27:33 -0700
Subject: [PATCH] Update build.Dockerfile

---
 ci/wheel/build.Dockerfile | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/ci/wheel/build.Dockerfile b/ci/wheel/build.Dockerfile
index c9f50514..8ff0b619 100644
--- a/ci/wheel/build.Dockerfile
+++ b/ci/wheel/build.Dockerfile
@@ -1,6 +1,5 @@
-ARG JOB_NAME
-ARG BUILD_NUM
-FROM rocm/pytorch-private:manylinux-hipclang-${JOB_NAME}-${BUILD_NUM}
+ARG DOCKER_TAG
+FROM ${DOCKER_TAG}
 
 RUN yum install -y python3-devel migraphx-devel
 RUN pip3 install pybind11-global