python - 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:")

标签： pythoninputprocessmpi