triton.language.program_id

triton.language.program_id(axis)

返回当前程序实例沿给定axis轴的ID。

Parameters:

axis (int) – 3D启动网格的轴。必须为0、1或2。