未验证 提交 38ff4fee 编写于 作者: L LiYuRio 提交者: GitHub

fix dygraph to dynamic (#54633)

上级 1f3dd978
...@@ -184,7 +184,7 @@ def _partial_send_op(tensor, group, dst, nranks, rank_id): ...@@ -184,7 +184,7 @@ def _partial_send_op(tensor, group, dst, nranks, rank_id):
group is not None group is not None
), "Group should be an instance for _partial_send_op." ), "Group should be an instance for _partial_send_op."
dst_rank_in_group = group.get_group_rank(dst) dst_rank_in_group = group.get_group_rank(dst)
if framework.in_dygraph_mode(): if framework.in_dynamic_mode():
return group.process_group.send_partial( return group.process_group.send_partial(
tensor, dst_rank_in_group, nranks, rank_id tensor, dst_rank_in_group, nranks, rank_id
) )
...@@ -195,7 +195,7 @@ def _partial_recv_op(tensor, group, src, nranks, rank_id): ...@@ -195,7 +195,7 @@ def _partial_recv_op(tensor, group, src, nranks, rank_id):
group is not None group is not None
), "Group should be an instance for _partial_recv_op." ), "Group should be an instance for _partial_recv_op."
src_rank_in_group = group.get_group_rank(src) src_rank_in_group = group.get_group_rank(src)
if framework.in_dygraph_mode(): if framework.in_dynamic_mode():
return group.process_group.recv_partial( return group.process_group.recv_partial(
tensor, src_rank_in_group, nranks, rank_id tensor, src_rank_in_group, nranks, rank_id
) )
...@@ -236,7 +236,7 @@ def partial_batch_isend_irecv(p2p_op_list): ...@@ -236,7 +236,7 @@ def partial_batch_isend_irecv(p2p_op_list):
if _warn_cur_rank_not_in_group(group): if _warn_cur_rank_not_in_group(group):
return return
if framework.in_dygraph_mode(): if framework.in_dynamic_mode():
group = _get_global_group() if group is None else group group = _get_global_group() if group is None else group
backend = group.backend backend = group.backend
tasks = [] tasks = []
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册