triton.language.program_id

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

返回当前程序实例在给定 axis（轴）上的 ID。

参数:

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