ProcessGrid#

class nvmath.distributed.distribution.ProcessGrid( *, shape: Sequence[int] | None = None, layout: GridLayout | None = None, process_array=None, )[source]#

N-dimensional grid of processes used by some distributions like the PBLAS block-cyclic distribution.

Example 2D process grid for 4 processes, with processes arranged in column-major order:

---------
| 0 | 2 |
---------
| 1 | 3 |
---------

Attributes

layout#: Layout of process grid if row-major or column-major, otherwise None.

process_array#

shape#: Shape of process grid.

Methods