diff --git a/docker/rocm/Dockerfile b/docker/rocm/Dockerfile index 94ee5481a..d55eacfcf 100644 --- a/docker/rocm/Dockerfile +++ b/docker/rocm/Dockerfile @@ -78,6 +78,8 @@ ENV MIGRAPHX_DISABLE_MIOPEN_FUSION=1 ENV MIGRAPHX_DISABLE_SCHEDULE_PASS=1 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 COPY --from=rocm-dist / /