From 0deefc37cceba661be69ebe8061a7451519e2d61 Mon Sep 17 00:00:00 2001 From: Tristan Rice Date: Wed, 29 Jan 2025 13:48:42 -0800 Subject: [PATCH] process_group: fix docs with torch==2.6.0 --- torchft/process_group.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/torchft/process_group.py b/torchft/process_group.py index d1d2cbe6..974f1b4e 100644 --- a/torchft/process_group.py +++ b/torchft/process_group.py @@ -132,10 +132,20 @@ def allgather( input_tensor: List[torch.Tensor], opts: object, ) -> Work: + """ + Gathers tensors from the whole group in a list. + + See torch.distributed.all_gather for more details. + """ raise NotImplementedError("not implemented") # pyre-fixme[14]: inconsistent override def broadcast(self, tensor_list: List[torch.Tensor], opts: object) -> Work: + """ + Broadcasts the tensor to the whole group. + + See torch.distributed.broadcast for more details. + """ raise NotImplementedError("not implemented") def broadcast_one(self, tensor: torch.Tensor, root: int) -> Work: