fix dygraph to dynamic (#54633)

PaddlePaddle · Jun 15, 2023 · 38ff4fe · 38ff4fe
1 parent 1f3dd97
commit 38ff4fe
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/paddle/distributed/fleet/meta_parallel/pp_utils/p2p_communication.py b/python/paddle/distributed/fleet/meta_parallel/pp_utils/p2p_communication.py
@@ -184,7 +184,7 @@ def _partial_send_op(tensor, group, dst, nranks, rank_id):
         group is not None
     ), "Group should be an instance for _partial_send_op."
     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(
             tensor, dst_rank_in_group, nranks, rank_id
         )
@@ -195,7 +195,7 @@ def _partial_recv_op(tensor, group, src, nranks, rank_id):
         group is not None
     ), "Group should be an instance for _partial_recv_op."
     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(
             tensor, src_rank_in_group, nranks, rank_id
         )
@@ -236,7 +236,7 @@ def partial_batch_isend_irecv(p2p_op_list):
     if _warn_cur_rank_not_in_group(group):
         return
 
-    if framework.in_dygraph_mode():
+    if framework.in_dynamic_mode():
         group = _get_global_group() if group is None else group
         backend = group.backend
         tasks = []