MPI Python输入函数，我的代码有问题

Question

我无法在每次处理后输入。执行mpiexec -n 4 python 1.py时获得的输出：

rank 0 hi
Enter
rank 1 hi
Enter
rank 2 hi
Enter
rank 3 hi
Enter

我只能输入一次输入，即在最后一个过程中，在执行每个等级或过程后如何输入

from mpi4py import MPI

comm = MPI.COMM_WORLD
rank = comm.rank
p=comm.Get_size()


if rank == 0:
    data = "hi
else:
    data = None

data = comm.bcast(data, root=0)

print('rank',rank,data)

x=input("Enter:")