From c2fa87a84e02b591697519a1ec87565342d4e706 Mon Sep 17 00:00:00 2001 From: youkaichao Date: Sun, 10 Nov 2024 17:39:07 -0800 Subject: [PATCH] fix code Signed-off-by: youkaichao --- docs/source/getting_started/debugging.rst | 3 +++ 1 file changed, 3 insertions(+) diff --git a/docs/source/getting_started/debugging.rst b/docs/source/getting_started/debugging.rst index 91978065faf4..d40222bfd4da 100644 --- a/docs/source/getting_started/debugging.rst +++ b/docs/source/getting_started/debugging.rst @@ -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