From bc22e1de9e3734d264933eb3b14301768ac7eb92 Mon Sep 17 00:00:00 2001 From: Khoa Pham Date: Mon, 21 Sep 2026 23:05:45 -0700 Subject: [PATCH] [DSpark] Fix draft CUDA graph stream explosion (#40658) --- python/sglang/srt/distributed/parallel_state.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/python/sglang/srt/distributed/parallel_state.py b/python/sglang/srt/distributed/parallel_state.py index 873633b1c..539a4f184 100644 --- a/python/sglang/srt/distributed/parallel_state.py +++ b/python/sglang/srt/distributed/parallel_state.py @@ -1518,9 +1518,7 @@ class GroupCoordinator: if self.world_size == 1: return input_ - # Always use pynccl to avoid capturing hip graph failure on torch - # version smaller than or equal to 2.11 - if is_hip() and self.pynccl_comm is not None and not self.pynccl_comm.disabled: + if self.pynccl_comm is not None and not self.pynccl_comm.disabled: self.pynccl_comm.broadcast(input_, src=src) else: # Broadcast.