Struct RandomPlan 

pub struct RandomPlan<T: Element, const N: usize> { /* private fields */ }

Random-generator plan.

Generic on T so the same type can carry the FP generators (RandomPlan<f32, N>, RandomPlan<f64, N>) and the Bernoulli generator (RandomPlan<Bool, N>). The element kind is reasserted in select() against the descriptor.

The plan owns a single cuRAND generator handle, created lazily on the first call to run (or any of the typed run_* accessors). cuRAND generators are not thread-safe; the plan is !Sync and !Send as a consequence (the Cell<curandGenerator_t> makes both negative).

Implementations

impl<T: Element, const N: usize> RandomPlan<T, N>

pub fn select( _stream: &Stream, desc: &RandomDescriptor<N>, _pref: PlanPreference, ) -> Result<Self>

Pick a kernel + validate the descriptor.

pub fn workspace_size(&self) -> usize

Workspace size in bytes.

Bernoulli needs numel * sizeof(f32) bytes for the uniform-rand intermediate buffer cuRAND writes into. Uniform / Normal need zero — cuRAND writes directly to the caller-provided output.

pub fn sku(&self) -> KernelSku

Kernel SKU identity.

pub fn precision_guarantee(&self) -> PrecisionGuarantee

Numerical guarantees.

impl<const N: usize> RandomPlan<f32, N>

pub fn run( &self, stream: &Stream, _workspace: Workspace<'_>, args: RandomArgs<'_, f32, N>, ) -> Result<()>

Generate Uniform(low, high) or Normal(mean, stddev) f32 samples.

impl<const N: usize> RandomPlan<f64, N>

pub fn run( &self, stream: &Stream, _workspace: Workspace<'_>, args: RandomArgs<'_, f64, N>, ) -> Result<()>

Generate Uniform(low, high) or Normal(mean, stddev) f64 samples.

impl<const N: usize> RandomPlan<Bool, N>

pub fn run( &self, stream: &Stream, workspace: Workspace<'_>, args: RandomBoolArgs<'_, N>, ) -> Result<()>

Generate a Bernoulli(p) sample tensor.

Trait Implementations

impl<T: Element, const N: usize> Drop for RandomPlan<T, N>

fn drop(&mut self)

Executes the destructor for this type.

fn pin_drop(self: Pin<&mut Self>)

🔬This is a nightly-only experimental API. (pin_ergonomics)

Execute the destructor for this type, but different to Drop::drop, it requires self to be pinned.