From 512d5f61903ffa7507f34762e052347ea6f6bc73 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Mon, 2 Jun 2025 08:36:36 -0600 Subject: [PATCH] Provide better support for running embeddings on GPU --- docker/rocm/Dockerfile | 1 + 1 file changed, 1 insertion(+) diff --git a/docker/rocm/Dockerfile b/docker/rocm/Dockerfile index da4774dfd..7cac69eef 100644 --- a/docker/rocm/Dockerfile +++ b/docker/rocm/Dockerfile @@ -63,6 +63,7 @@ COPY --from=rocm /opt/rocm-dist/ / FROM deps-prelim AS rocm-prelim-hsa-override0 ENV HSA_ENABLE_SDMA=0 ENV MIGRAPHX_ENABLE_NHWC=1 +ENV TF_ROCM_USE_IMMEDIATE_MODE=1 COPY --from=rocm-dist / /