From 38ff4feee5cacd5446e1ae46a96dd8ef34f14bef Mon Sep 17 00:00:00 2001
From: LiYuRio <63526175+LiYuRio@users.noreply.github.com>
Date: Thu, 15 Jun 2023 13:30:20 +0800
Subject: [PATCH] fix dygraph to dynamic (#54633)

---
 .../fleet/meta_parallel/pp_utils/p2p_communication.py       | 6 +++---
 1 file changed, 3 insertions(+), 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
index c2e8f83e93b..eede472f792 100644
--- 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 = []
-- 
GitLab