Log error and exit if MPI_COMM_WORLD is too small

这个提交包含在:
Nathan Mannall
2025-07-17 16:14:55 +01:00
父节点 6805dee72e
当前提交 db00d14f2c

查看文件

@@ -169,10 +169,11 @@ class MPIContext(Context):
requested_mpi_size = np.prod(config.sim_config.mpi)
if self.comm.size < requested_mpi_size:
raise ValueError(
logger.error(
f"MPI_COMM_WORLD size of {self.comm.size} is too small for requested dimensions of"
f" {config.sim_config.mpi}. {requested_mpi_size} ranks are required."
)
exit()
if self.rank >= requested_mpi_size:
logger.warning(