Unverified Commit f0f2e563 authored by youkaichao's avatar youkaichao Committed by GitHub
Browse files

[doc] improve debugging code (#10206)


Signed-off-by: default avataryoukaichao <youkaichao@gmail.com>
parent ad9a78bf
......@@ -75,6 +75,9 @@ If GPU/CPU communication cannot be established, you can use the following Python
print("PyTorch GLOO is successful!")
if world_size <= 1:
exit()
# Test vLLM NCCL, with cuda graph
from vllm.distributed.device_communicators.pynccl import PyNcclCommunicator
......
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