Slab

enum nvmath.distributed.fft.Slab(value)

An IntEnum class to specify a cuFFTMp Slab distribution.

Given an array of size X * Y * Z distributed over n GPUs, there are two possible slab distributions depending on whether the data is partitioned on the X or Y axis:

• X axis partitioning: the first X % n GPUs each own (X/n+1) * Y * Z elements and the remaining GPUs each own (X/n) * Y * Z elements.

• Y axis partitioning: the first Y % n GPUs each own X * (Y/n+1) * Z elements and the remaining GPUs each own X * (Y/n) * Z elements.

See also

FFT, fft()

Member Type:

int

Valid values are as follows:

X = <Slab.X: 2>

Y = <Slab.Y: 3>