Unverified Commit 871d6b7c authored by Lu Fang's avatar Lu Fang Committed by GitHub
Browse files

[Misc] Reduce warning message introduced in env_override (#19476)


Signed-off-by: default avatarLu Fang <lufang@fb.com>
parent 29a38f03
...@@ -13,7 +13,7 @@ logger = init_logger(__name__) ...@@ -13,7 +13,7 @@ logger = init_logger(__name__)
# that interact with vllm workers. # that interact with vllm workers.
# they are executed whenever `import vllm` is called. # they are executed whenever `import vllm` is called.
if 'NCCL_CUMEM_ENABLE' in os.environ: if os.environ.get('NCCL_CUMEM_ENABLE', '0') != '0':
logger.warning( logger.warning(
"NCCL_CUMEM_ENABLE is set to %s, skipping override. " "NCCL_CUMEM_ENABLE is set to %s, skipping override. "
"This may increase memory overhead with cudagraph+allreduce: " "This may increase memory overhead with cudagraph+allreduce: "
......
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