Skip to content

Commit 4bdb8a7

Browse files
authored
process_group: fix docs with torch==2.6.0 (#88)
1 parent 866873a commit 4bdb8a7

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

torchft/process_group.py

+10
Original file line numberDiff line numberDiff line change
@@ -132,10 +132,20 @@ def allgather(
132132
input_tensor: List[torch.Tensor],
133133
opts: object,
134134
) -> Work:
135+
"""
136+
Gathers tensors from the whole group in a list.
137+
138+
See torch.distributed.all_gather for more details.
139+
"""
135140
raise NotImplementedError("not implemented")
136141

137142
# pyre-fixme[14]: inconsistent override
138143
def broadcast(self, tensor_list: List[torch.Tensor], opts: object) -> Work:
144+
"""
145+
Broadcasts the tensor to the whole group.
146+
147+
See torch.distributed.broadcast for more details.
148+
"""
139149
raise NotImplementedError("not implemented")
140150

141151
def broadcast_one(self, tensor: torch.Tensor, root: int) -> Work:

0 commit comments

Comments
 (0)