INNER CODE UNIT · Rust

set_seed

NVlabs/cuda-oxide · cuda-bindings/src/lib.rs:240

    unsafe fn set_seed(gen: curandGenerator_t, seed: u64) {
        assert!(curandSetPseudoRandomGeneratorSeed(gen, c_ulonglong::from(seed)) == 0);
    }

    #[test]
    fn cu_event_elapsed_time_helper_signature() {
        // Compile-time check that the helper keeps the signature cuda-oxide's
        // bindings expose.
        let _: unsafe fn(*mut f32, CUevent, CUevent) -> CUresult = cu_event_elapsed_time;
    }

    #[test]
    fn test_curand() {
        unsafe {
            let (_dev, _ctx) = init();
            let curand_gen = get_rng();
            set_seed(curand_gen, 123);
            let num_elements = 32;

View source record →