INNER CODE UNIT · Rust
test_curand
NVlabs/cuda-oxide · cuda-bindings/src/lib.rs:252
fn test_curand() {
unsafe {
let (_dev, _ctx) = init();
let curand_gen = get_rng();
set_seed(curand_gen, 123);
let num_elements = 32;
let bytesize = num_elements * size_of::<f32>();
let dptr = get_dptr(bytesize);
assert!(
curandGenerateNormal(curand_gen, dptr as *mut f32, num_elements, 0.0, 1.0) == 0
);
assert!(curandDestroyGenerator(curand_gen) == 0);
assert!(cuMemFree_v2(dptr) == 0);
}
}
}