如果處理器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()