triton.language.randint

triton.language.randint(seed, offset, n_rounds: constexpr = 10)

Given a seed scalar and an offset block, returns a single block of random int32.

If you need multiple streams of random numbers, using randint4x is likely to be faster than calling randint 4 times.

Parameters:
  • seed – The seed for generating random numbers.

  • offset – The offsets to generate random numbers for.