mpi4py.MPI.Compute_dims

mpi4py.MPI.Compute_dims(nnodes, dims)

Return a balanced distribution of processes per coordinate direction

Parameters
  • nnodes (int) –

  • dims (Union[int, Sequence[int]]) –

Return type

List[int]