An op which returns the number of MPI processes. This is equivalent to running `MPI_Comm_size(MPI_COMM_WORLD, ...)` to get the size of the global communicator. Returns: An integer scalar containing the number of MPI processes.
(name=None)
| 59 | |
| 60 | |
| 61 | def size(name=None): |
| 62 | """An op which returns the number of MPI processes. |
| 63 | |
| 64 | This is equivalent to running `MPI_Comm_size(MPI_COMM_WORLD, ...)` to get the |
| 65 | size of the global communicator. |
| 66 | |
| 67 | Returns: |
| 68 | An integer scalar containing the number of MPI processes. |
| 69 | """ |
| 70 | return MPI_LIB.mpi_size(name=name) |
| 71 | |
| 72 | |
| 73 | ops.NotDifferentiable('MPISize') |
no outgoing calls
no test coverage detected