diff --git a/docker/Dockerfile.rocm b/docker/Dockerfile.rocm index af9f9e24d..e1a242c87 100644 --- a/docker/Dockerfile.rocm +++ b/docker/Dockerfile.rocm @@ -58,6 +58,7 @@ RUN git clone ${ATER_REPO} \ # Performance environment variable. ENV HIP_FORCE_DEV_KERNARG=1 +ENV HSA_NO_SCRATCH_RECLAIM=1 ENV SGLANG_SET_CPU_AFFINITY=1 ENV SGLANG_ALLOW_OVERWRITE_LONGER_CONTEXT_LEN=1 ENV NCCL_MIN_NCHANNELS=112