скрипт python работает с mpirun не останавливается, если assert на процессоре 0 не удается

Question

У меня есть скрипт python с набором операций, выполняемых параллельно, с библиотекой mpi4py. В конце операций процессор с рангом 0 выполняет тест assert. Если утверждение не выполняется, процесс должен остановиться, и программа завершится. Однако программа не выходит, и я думаю, это потому, что другие процессоры держатся. Как заставить программу завершить выполнение, если утверждение не выполнено? я бегу вещи с помощью следующей команды:

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()