triton.language.arange

triton.language.arange(start, end)

Returns contiguous values within the half-open interval [start, end). end - start must be less than or equal to TRITON_MAX_TENSOR_NUMEL = 1048576

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 greater than start.