|
|
|
@@ -304,7 +304,7 @@ class HostReduceScatter(PrimitiveWithInfer): |
|
|
|
like sum, max, avg. Default: ReduceOp.SUM. |
|
|
|
group (Union[tuple[int],list[int]]): The rand_ids of communication group to work on. |
|
|
|
|
|
|
|
Raise: |
|
|
|
Raises: |
|
|
|
TypeError: If op is not a string and group is not a list nor tuple, |
|
|
|
or elements of group are not int. |
|
|
|
ValueError: If the first dimension of input can not be divided by rank size, |
|
|
|
|