This document is relevant for: Inf2, Trn1, Trn1n

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

This document is relevant for: Inf2, Trn1, Trn1n