ROCm Container: set SGLANG_SET_CPU_AFFINITY=1 (#2328)

This commit is contained in:
HAI
2024-12-02 23:20:33 -08:00
committed by GitHub
parent aa47f64223
commit 0639bf15d1

View File

@@ -33,6 +33,7 @@ RUN python -m pip cache purge
# Performance environment variable.
ENV HIP_FORCE_DEV_KERNARG=1
ENV SGLANG_SET_CPU_AFFINITY=1
ENV SGLANG_ALLOW_OVERWRITE_LONGER_CONTEXT_LEN=1
ENV NCCL_MIN_NCHANNELS=112