Revert the changes on NCCL symmetric memory (#10210)

Co-authored-by: Yineng Zhang <me@zhyncs.com>
2025-09-09 11:01:33 -07:00
parent d3ee70985f
commit d352c29aa0
5 changed files with 7 additions and 43 deletions
--- a/python/sglang/srt/distributed/parallel_state.py
+++ b/python/sglang/srt/distributed/parallel_state.py
@@ -510,17 +510,6 @@ class GroupCoordinator:
        if self.npu_communicator is not None and not self.npu_communicator.disabled:
            return self.npu_communicator.all_reduce(input_)

-        if (
-            self.pynccl_comm is not None
-            and hasattr(input_, "symmetric_memory")
-            and input_.symmetric_memory
-        ):
-            with self.pynccl_comm.change_state(
-                enable=True, stream=torch.cuda.current_stream()
-            ):
-                self.pynccl_comm.all_reduce(input_)
-                return input_
-
        outplace_all_reduce_method = None
        if (
            self.qr_comm is not None