diff --git a/docker/rocm/Dockerfile b/docker/rocm/Dockerfile index d55eacfcf..653ed1e4e 100644 --- a/docker/rocm/Dockerfile +++ b/docker/rocm/Dockerfile @@ -80,6 +80,8 @@ ENV MIGRAPHX_DISABLE_REDUCE_FUSION=1 ENV MIGRAPHX_ENABLE_HIPRTC_WORKAROUNDS=1 ENV MIOPEN_CUSTOM_CACHE_DIR=/config/model_cache/migraphx ENV MIOPEN_USER_DB_PATH=/config/model_cache/migraphx +ENV AMD_COMGR_CACHE=1 +ENV AMD_COMGR_CACHE_DIR=/config/model_cache/migraphx COPY --from=rocm-dist / /