Fix merge bug

Author: waliwali777

python/paddle/distributed/auto_parallel/static/reshard_funcs/same_status_reshard_func.py

@@ -100,7 +100,7 @@ def reshard(self, src_dist_attr, dst_dist_attr, src_value, dst_type):
                 paddle._C_ops.p_send(
                     src_value,
                     comm_group.id,
-                    dst_local_rank,
+                    comm_group.ranks.index(dst),
                     False,
                 )
                 point = paddle.base.libpaddle.pir.get_current_insertion_point()
@@ -128,7 +128,7 @@ def reshard(self, src_dist_attr, dst_dist_attr, src_value, dst_type):
                 comm_group = new_process_group([src, dst], group_type="p2p")
                 recv_value = paddle._C_ops.p_recv(
                     comm_group.id,
-                    src_local_rank,
+                    comm_group.ranks.index(src),
                     dst_type.dtype,
                     False,
                 )