Log error before calling MPI abort

这个提交包含在:
nmannall
2025-01-21 15:20:03 +00:00
父节点 d162365df7
当前提交 897aa1d252

查看文件

@@ -188,6 +188,7 @@ class MPIContext(Context):
try:
return super().run()
except:
logger.exception(f"Rank {self.rank} encountered an error. Aborting...")
self.comm.Abort()
def _run_model(self, model_num: int) -> None: