nki.isa.set_rng_seed#

nki.isa.set_rng_seed(src_seeds, name=None)[source]#

Seed the pseudo random number generator (PRNG) inside the Vector Engine.

The PRNG state is cached inside the engine as a persistent state during the rest of NEFF execution. However, the state cannot survive TPB resets or Runtime reload.

Using the same seed will generate the same sequence of random numbers when used together with the nisa.rng() on the Vector Engine.

Memory types.

The input src_seeds must be in SBUF or PSUM.

Data types.

The input src_seeds must be a 32-bit value.

Tile size.

The input src_seeds must be a [1,1] tensor.

Parameters:

src_seeds – a [1,1] tensor on SBUF or PSUM with a 32-bit value to be used as the seed