This document is relevant for: Trn2, Trn3
nki.language.num_programs#
- nki.language.num_programs(axes=0)[source]#
Number of SPMD programs along the given axes in the launch grid.
- Parameters:
axes – the axes of the launch grid. If not provided, returns the total number of programs along the entire launch grid.
- Returns:
the number of SPMD programs along
axesin the launch grid.
This document is relevant for: Trn2, Trn3