Skip to content

P
Paddle

PaddlePaddle / Paddle
1 年多 前同步成功

通知 2302
Star 20931
Fork 5422
P
Paddle
未验证 提交 e6e62342 编写于 作者: T Tian 提交者： GitHub
浏览文件
操作

bugfix on dist.alltoall_single (#52495)

上级 7ee31e72
显示空白变更内容
内联 并排
Showing with 1 addition and 1 deletion
python/paddle/distributed/communication/stream/all_to_all.py
浏览文件 @ e6e62342
...@@ -220,7 +220,7 @@ def _alltoall_single_in_dygraph( ...@@ -220,7 +220,7 @@ def _alltoall_single_in_dygraph(
sync_op, sync_op,
use_calc_stream, use_calc_stream,
): ):
world_size = dist.get_world_size() world_size = dist.get_world_size(group)
if out_split_sizes is None: if out_split_sizes is None:
out_split_sizes = [ out_split_sizes = [
out_tensor.shape[0] // world_size for _ in range(world_size) out_tensor.shape[0] // world_size for _ in range(world_size)
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或