nki.language.program_ndim

nki.language.program_ndim()

Number of dimensions in the SPMD launch grid.

Returns:

The number of dimensions in the launch grid, i.e. the number of axes