triton.language.rand

triton.language.rand = JITFunction(triton.language.random:rand)

Given a seed scalar and an offset block, returns a block of random float32 in \(U(0, 1)\).

Parameters:

• seed – The seed for generating random numbers.

• offsets – The offsets to generate random numbers for.