tilelang.language.random

Functions

rng_init(seed[, seq, off])	Initialize CUDA curand random number generator state
rng_rand()	Generate a 32-bit unsigned random integer

Module Contents

tilelang.language.random.rng_init(seed, seq=None, off=0)

Initialize CUDA curand random number generator state

Parameters:

• seed (PrimExpr) – Random seed value.

• seq (PrimExpr) – Sequence number for parallel random number generation.

• off (PrimExpr) – Offset number for parallel random number generation.

Returns:

state – The random number generator state handle.

Return type:

PrimExpr

tilelang.language.random.rng_rand()

Generate a 32-bit unsigned random integer

Returns:

random_value – A 32-bit unsigned random integer.

Return type:

PrimExpr