Skip to content

Commit

Permalink
[DTensor][Easy] Minor fix to Partial Placement Docstring (pytorch#133149
Browse files Browse the repository at this point in the history
)

Minor doc fix: The reduce op string for product should be "product" instead of "prod".
https://github.com/pytorch/pytorch/blob/main/torch/distributed/_functional_collectives.py#L1045

Pull Request resolved: pytorch#133149
Approved by: https://github.com/XilunWu, https://github.com/tianyu-l
  • Loading branch information
wz337 authored and pytorchmergebot committed Aug 15, 2024
1 parent e6272ac commit cf1fc07
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torch/distributed/_tensor/placement_types.py
Original file line number Diff line number Diff line change
Expand Up @@ -572,7 +572,7 @@ class Partial(Placement):
Args:
reduce_op (str, optional): The reduction op to be used for the partial DTensor
to produce Replicated/Sharded DTensor. Only element-wise reduction operations
are supportd, including: "sum", "avg", "prod", "max", "min", default: "sum".
are supportd, including: "sum", "avg", "product", "max", "min", default: "sum".
::note:: The ``Partial`` placement can be generated as a result of the DTensor operators,
and can only be used by the ``DTensor.from_local`` API.
Expand Down

0 comments on commit cf1fc07

Please sign in to comment.