fix bug of p2p (#33929)

Author: ForFishes

python/paddle/distributed/fleet/meta_parallel/pp_utils/p2p_communication.py

@@ -13,15 +13,17 @@
 # limitations under the License.
 
 import paddle
-import paddle.distributed as dist
 
 _groups = None
 _hcg = None
 
 
 def initialize_p2p_groups(hcg):
     global _groups, _hcg
-    _groups = [dist.new_group(ranks=group) for group in hcg.get_p2p_groups()]
+    _groups = [
+        paddle.distributed.new_group(ranks=group)
+        for group in hcg.get_p2p_groups()
+    ]
     _hcg = hcg
 
 
@@ -33,7 +35,7 @@ def send(tensor, dest_stage):
     _is_valid_communciate(src_stage, dest_stage)
     group = _get_send_recv_group(src_stage, dest_stage)
     dst_rank = _hcg.get_rank_from_stage(stage_id=dest_stage)
-    return dist.broadcast(tensor, src_rank, group=group)
+    return paddle.distributed.broadcast(tensor, src_rank, group=group)
 
 
 def recv(tensor, src_stage):
@@ -43,7 +45,7 @@ def recv(tensor, src_stage):
     _is_valid_communciate(src_stage, dest_stage)
     group = _get_send_recv_group(src_stage, dest_stage)
     src_rank = _hcg.get_rank_from_stage(stage_id=src_stage)
-    return dist.broadcast(tensor, src_rank, group=group)
+    return paddle.distributed.broadcast(tensor, src_rank, group=group)
 
 
 def _is_valid_communciate(src_stage, dest_stage):