Add ROCM_PATH

ROCm · Jul 15, 2024 · c7007f7 · c7007f7
1 parent 22dabc1
commit c7007f7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/manywheel/Dockerfile_2_28 b/manywheel/Dockerfile_2_28
@@ -136,6 +136,9 @@ FROM common as rocm_final
 ARG ROCM_VERSION=6.0
 ARG PYTORCH_ROCM_ARCH
 ENV PYTORCH_ROCM_ARCH ${PYTORCH_ROCM_ARCH}
+# Somewhere in ROCm stack, we still use non-existing /opt/rocm/hip path, 
+# below workaround helps avoid error
+ENV ROCM_PATH /opt/rocm
 ADD ./common/install_rocm_drm.sh install_rocm_drm.sh
 RUN bash ./install_rocm_drm.sh && rm install_rocm_drm.sh
 # cmake3 is needed for the MIOpen build