update

pytorch · Jul 20, 2023 · b866964 · b866964
1 parent e2b5b90
commit b866964
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch/nn/parallel/comm.py b/torch/nn/parallel/comm.py
@@ -162,7 +162,7 @@ def scatter(tensor, devices=None, chunk_sizes=None, dim=0, streams=None, *, out=
           into equal chunks.
         dim (int, optional): A dimension along which to chunk :attr:`tensor`.
           Default: ``0``.
-        streams (Iterable[Stream], optional): an iterable of Streams, among
+        streams (Iterable[torch.cuda.Stream], optional): an iterable of Streams, among
           which to execute the scatter. If not specified, the default stream will
           be utilized.
         out (Sequence[Tensor], optional, keyword-only): the GPU tensors to