[DSpark] Fix draft CUDA graph stream explosion (#40658)

2026-09-21 23:05:45 -07:00
parent a0781f2714
commit bc22e1de9e
1 changed files with 1 additions and 3 deletions
@@ -1518,9 +1518,7 @@ class GroupCoordinator:
        if self.world_size == 1:
            return input_
-        # Always use pynccl to avoid capturing hip graph failure on torch
+        if self.pynccl_comm is not None and not self.pynccl_comm.disabled:
        # version smaller than or equal to 2.11
        if is_hip() and self.pynccl_comm is not None and not self.pynccl_comm.disabled:
            self.pynccl_comm.broadcast(input_, src=src)
        else:
            # Broadcast.