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

Question

I have a python script with a set of operations done in parallel, with the library mpi4py. At the end of the operations the processor with rank 0 executes an assert test. If the assert fails, the process should stop and the program terminate. However, the program doesn't exit and this I guess is because the other processors are holding. How to make the program ending the execution if the assert fails? I run things with a command like:

mpirun -np 10 python myscript.py 

and then I have a line in the code like:

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

Answer 1

Instead of assert, you should abort.

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

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