triton.language.num_programs

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

Returns the number of program instances launched along the given axis.

Parameters

• axis (int) – The axis of the 3D launch grid. Has to be either 0, 1 or 2.

• builder (triton.ir.builder, optional from within JIT'ed functions) – IR builder to generate code into