зеркало из https://github.com/microsoft/DeepSpeed.git
single node pdsh sigkill (#3730)
Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
This commit is contained in:
Родитель
8bfbb0e3ca
Коммит
26b3e73298
|
@ -564,7 +564,7 @@ def main(args=None):
|
|||
time.sleep(1)
|
||||
sys.exit(1)
|
||||
|
||||
if args.launcher == PDSH_LAUNCHER:
|
||||
if args.launcher == PDSH_LAUNCHER and multi_node_exec:
|
||||
signal.signal(signal.SIGINT, sigkill_handler)
|
||||
|
||||
result.wait()
|
||||
|
|
Загрузка…
Ссылка в новой задаче