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.