triton.language.program_id

triton.language.program_id(axis, _builder=None)

Returns the id of the current program instance along the given axis.

参数:

axis (int) -- The axis of the 3D launch grid. Must be 0, 1 or 2.