MPI信號處理

Question

使用mpirun時，是否可以在正在運行的代碼中捕獲信號（例如，由^C生成的SIGINT）？

例如，我正在運行一個並行化的Python代碼。我可以用except KeyboardInterrupt自己運行python blah.py來捕捉這些錯誤，但我不能在做mpirun -np 1 python blah.py時。

有沒有人有建議？即使找到如何在C或C++編譯的程序中捕捉信號也是一個有益的開始。

如果我向派生的Python進程發送信號，他們可以正確處理信號;然而，發送到父節點的信號（即，超過羣集上的掛機時間，或者在終端中按下control-C）將立即終止一切。

Answer 1

的signal模塊支持使用signal.signal設置信號處理程序：

Set the handler for signal signalnum to the function handler. handler can be a callable Python object taking two arguments (see below), or one of the special values signal.SIG_IGN or signal.SIG_DFL. The previous signal handler will be returned ...

import signal 
def ignore(sig, stack): 
    print "I'm ignoring signal %d" % (sig,) 

signal.signal(signal.SIGINT, ignore) 
while True: pass 

如果發送SIGINT運行此腳本Python解釋器（通過kill -INT <pid>），它將打印一條消息，簡單地繼續跑。

Answer 2

如果您使用mpirun --nw，那麼mpirun本身應該在其開始子流程時立即終止，而不是等待它們的終止;如果這是可以接受的，那麼我相信你的程序將能夠捕捉到他們自己的信號。