triton.language.arange

triton.language.arange(start, end)

Returns contiguous values within the left-closed and right-open interval [start, end). End - Start must be less than or equal to TRITON_MAX_TENSOR_NUMEL = 131072

Parameters:

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

  • end (int32) – End of the interval. Must be a power of two > start.