Skip to content

GitLab

Unverified Commit 17dbf976 authored by HAI's avatar HAI Committed by GitHub
Browse files

update ENV to ROCm dockers (#3248)

parent 53179026
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
docker/Dockerfile.rocm
View file @ 17dbf976
...@@ -58,6 +58,7 @@ RUN git clone ${ATER_REPO} \ ...@@ -58,6 +58,7 @@ RUN git clone ${ATER_REPO} \
# Performance environment variable. # Performance environment variable.
ENV HIP_FORCE_DEV_KERNARG=1 ENV HIP_FORCE_DEV_KERNARG=1
ENV HSA_NO_SCRATCH_RECLAIM=1
ENV SGLANG_SET_CPU_AFFINITY=1 ENV SGLANG_SET_CPU_AFFINITY=1
ENV SGLANG_ALLOW_OVERWRITE_LONGER_CONTEXT_LEN=1 ENV SGLANG_ALLOW_OVERWRITE_LONGER_CONTEXT_LEN=1
ENV NCCL_MIN_NCHANNELS=112 ENV NCCL_MIN_NCHANNELS=112
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment