triton.language.arange

triton.language.arange(start, end, builder=None)

Returns contiguous values within the open interval [start, end).

Parameters

• start (int) – Start of the interval. Must be a power of two.

• stop (int) – End of the interval. Must be a power of two >= start.

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