cubecl-random 0.8.1

CubeCL Random Number Generation.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

pub mod bernoulli;
pub mod interval;
pub mod normal;
pub mod uniform;

#[allow(missing_docs)]
#[macro_export]
macro_rules! testgen_random {
    () => {
        use cubecl::prelude::*;
        use cubecl_core as cubecl;

        use cubecl::{client::ComputeClient, prelude::TensorHandleRef};
        use cubecl_random::*;
        use cubecl_std::tensor::TensorHandle;

        use core::f32;

        cubecl_random::testgen_random_bernoulli!();
        cubecl_random::testgen_random_normal!();
        cubecl_random::testgen_random_uniform!();
        cubecl_random::testgen_random_interval!();
    };
}