This document is relevant for: Inf2, Trn1, Trn1n

nki.language.num_programs#

nki.language.num_programs(axes=None)[source]#

Number of SPMD programs along the given axes in the launch grid. If axes is not provided, returns the total number of programs.

Parameters:: axes – The axes of the ND launch grid. If not provided, returns the total number of programs along the entire launch grid.
Returns:: The number of SPMD(single process multiple data) programs along axes in the launch grid

This document is relevant for: Inf2, Trn1, Trn1n

nki.language.num_programs