如果处理器0上的断言失败，以mpirun运行的python脚本不会停止

Question

我有一个python脚本，带有一组并行执行的操作，以及库mpi4py。 在操作结束时，等级为0的处理器将执行断言测试。 如果断言失败，则该过程应停止并且程序终止。 但是，该程序没有退出，我想这是因为其他处理器正在处理。 如果断言失败，如何使程序结束执行？ 我使用以下命令运行内容：

mpirun -np 10 python myscript.py 

然后我在代码中一行：

if rank ==0:
    assert mytest()==0

Answer 1

而不是断言，您应该中止。

https://planet.scipy.org/docs/apiref/mpi4py.MPI.Comm-class.html#Abort

if rank == 0:
    if mytest() === 0:
        comm.Abort()