bugfix on dist.alltoall_single (#52495)

e6e62342 · Tian · GitHub · 7ee31e72 · e6e62342
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/distributed/communication/stream/all_to_all.py python/paddle/distributed/communication/stream/all_to_all.py +1 -1

未找到文件。
--- a/python/paddle/distributed/communication/stream/all_to_all.py
+++ b/python/paddle/distributed/communication/stream/all_to_all.py
@@ -220,7 +220,7 @@ def _alltoall_single_in_dygraph(
    sync_op,
    use_calc_stream,
 ):
-    world_size = dist.get_world_size()
+    world_size = dist.get_world_size(group)
    if out_split_sizes is None:
        out_split_sizes = [
            out_tensor.shape[0] // world_size for _ in range(world_size)