triton.language.num_programs

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

返回沿给定 axis 启动的程序实例数量。

参数:

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