chenchenghao/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 11 Packages Projects Releases Wiki Activity

Try to allow NCCL cumem for multi node nvlink case (#11987)

Browse Source
This commit is contained in:
fzyzcjy
2025-11-01 03:48:25 +08:00
committed by GitHub
parent a80bcb5a68
commit 30ad107028
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/sglang/srt/entrypoints/engine.py
View File

@@ -677,7 +677,8 @@ class Engine(EngineBase):
def _set_envs_and_config(server_args: ServerArgs):
# Set global environments
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
os.environ["NCCL_CUMEM_ENABLE"] = str(int(server_args.enable_symm_mem))
if "NCCL_CUMEM_ENABLE" not in os.environ:
os.environ["NCCL_CUMEM_ENABLE"] = str(int(server_args.enable_symm_mem))
if not server_args.enable_symm_mem:
os.environ["NCCL_NVLS_ENABLE"] = str(int(server_args.enable_nccl_nvls))
os.environ["CUDA_DEVICE_MAX_CONNECTIONS"] = "8"