tilelang.language.random

Functions

rng_init(seed[, seq, off, generator])

Initialize CUDA curand random number generator state

rng_rand()

Generate a 32-bit unsigned random integer

rng_rand_float([bit, dist])

Generate a random float

Module Contents

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

Initialize CUDA curand random number generator state

参数:

  • seed (PrimExpr) -- Random seed value.

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

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

  • generator (StringImm) -- Set random generator. See https://docs.nvidia.com/cuda/curand/group__DEVICE.html

返回:

state -- The random number generator state handle.

返回类型:

PrimExpr

tilelang.language.random.rng_rand()

Generate a 32-bit unsigned random integer

返回:

random_value -- A 32-bit unsigned random integer.

返回类型:

PrimExpr

tilelang.language.random.rng_rand_float(bit=32, dist='uniform')

Generate a random float

参数:

  • bit (int = [32, 64]) -- Bitwidth of random float.

  • dist (StringImm = ["uniform", "normal"]) -- Random distribution.

返回:

random_value -- A random float.

返回类型:

PrimExpr