This document is relevant for: Trn2, Trn3

nki.language.program_ndim#

nki.language.program_ndim()[source]#

Number of dimensions in the SPMD launch grid.

Returns:

the number of dimensions in the launch grid, i.e. the number of axes. 0 if no grid.

This document is relevant for: Trn2, Trn3