This document is relevant for: Trn2, Trn3
nki.language.program_id#
- nki.language.program_id(axis=0)[source]#
Index of the current SPMD program along the given axis in the launch grid.
- Parameters:
axis – the axis of the launch grid.
- Returns:
the program id along
axis.
This document is relevant for: Trn2, Trn3