Struct cudarc::curand::safe::CudaRng

pub struct CudaRng { /* private fields */ }

Host side RNG that can fill CudaSlice with random values.

1. Create:

let device = CudaDevice::new(0).unwrap();
let rng = CudaRng::new(0, device).unwrap();

2. Fill device memory:

let mut a_dev = device.alloc_zeros::<f32>(10).unwrap();
rng.fill_with_uniform(&mut a_dev).unwrap();

The three distributions are:

1. Uniform - CudaRng::fill_with_uniform()
2. Normal - CudaRng::fill_with_normal()
3. LogNormal - CudaRng::fill_with_log_normal()

Implementations

impl CudaRng

pub fn new(seed: u64, device: Arc<CudaDevice>) -> Result<Self, CurandError>

Constructs the RNG with the given seed. Requires the stream from CudaDevice to submit kernels.

pub fn set_seed(&mut self, seed: u64) -> Result<(), CurandError>

Re-seed the RNG.

pub fn fill_with_uniform<T>( &self, t: &mut CudaSlice<T> ) -> Result<(), CurandError>where curandGenerator_t: UniformFill<T>,

Fill the CudaSlice with data from a Uniform distribution

pub fn fill_with_normal<T>( &self, t: &mut CudaSlice<T>, mean: T, std: T ) -> Result<(), CurandError>where curandGenerator_t: NormalFill<T>,

Fill the CudaSlice with data from a Normal(mean, std) distribution.

pub fn fill_with_log_normal<T>( &self, t: &mut CudaSlice<T>, mean: T, std: T ) -> Result<(), CurandError>where curandGenerator_t: LogNormalFill<T>,

Fill the CudaRc with data from a LogNormal(mean, std) distribution.

Trait Implementations

impl Drop for CudaRng

fn drop(&mut self)

Executes the destructor for this type.