4.6.8. pde.tools.mpi module
Auxillary functions and variables for dealing with MPI multiprocessing
Warning
These functions are mostly no-ops unless MPI is properly installed and python code
was started using mpirun
or mpiexec
. Please refer to the
documentation of your MPI distribution for details.
send data to another MPI node |
|
receive data from another MPI node |
|
combines data from all MPI nodes |
- is_main: bool = True
Flag indicating whether the current process is the main process (with ID 0)
- Type:
- mpi_allreduce(data, operator=None)[source]
combines data from all MPI nodes
Note that complex datatypes and user-defined functions are not properly supported.
- Parameters:
data – Data being send from this node to all others
operator (int | Operator | None) – The operator used to combine all data. Possible options are summarized in the IntEnum
numba_mpi.Operator
.
- Returns:
The accumulated data
- ol_mpi_allreduce(data, operator=None)[source]
overload the mpi_allreduce function
- Parameters:
operator (int | Operator | None) –