mpiexec

nnodes.mpiexec._dispatch(lock: Lock, nnodes: fractions.Fraction | int) → bool

Execute a task if resource is available.

async nnodes.mpiexec.mpiexec(cmd: Union[Callable, List[str], Tuple[str, ...], str], nprocs: Union[int, Callable[[Directory], int]], cpus_per_proc: int, gpus_per_proc: int, mps: int | None, fname: str | None, args: list | tuple | None, mpiarg: list | tuple | None, group_mpiarg: bool, check_output: Optional[Callable[[...], None]], use_multiprocessing: bool | None, timeout: Optional[Union[Literal['auto'], float]], ontimeout: Optional[Union[Literal['raise'], Callable[[], None]]], d: Directory) → str

Schedule the execution of MPI task.

nnodes.mpiexec.splitargs(mpiarg: list | tuple, nprocs: int) → list

Split arguments to n processes.