Merge branch 'mpi' into 34-reframe-mpi-benchmarks

这个提交包含在:
nmannall
2024-11-19 12:06:07 +00:00
当前提交 42ee2ac990
共有 2 个文件被更改,包括 3 次插入1 次删除

查看文件

@@ -175,7 +175,7 @@ class MPIContext(Context):
raise ValueError
if self.rank >= requested_mpi_size:
logger.warn(
logger.warning(
f"Rank {self.rank}: Only {requested_mpi_size} MPI ranks required for the"
" dimensions specified. Either increase your MPI dimension size, or request"
" fewer MPI tasks."

查看文件

@@ -58,6 +58,8 @@ class MPIModel(Model):
return super().build_geometry()
def _broadcast_model(self):
self.title = self.comm.bcast(self.title)
self.nx = self.comm.bcast(self.nx)
self.ny = self.comm.bcast(self.ny)
self.nz = self.comm.bcast(self.nz)