Struct CastPlan 

pub struct CastPlan<TIn: Element, TOut: Element> { /* private fields */ }

Cast plan.

TIn is the input element type. TOut is the output element type.

Implementations

impl<TIn: Element, TOut: Element> CastPlan<TIn, TOut>

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

Pick a kernel for desc.

pub fn can_implement(&self, args: &CastArgs<'_, TIn, TOut>) -> Result<()>

Validate args.

pub fn workspace_size(&self) -> usize

Workspace size in bytes. Always 0.

pub fn sku(&self) -> KernelSku

Identity of the kernel this plan picked.

pub fn precision_guarantee(&self) -> PrecisionGuarantee

Numerical guarantees for this plan’s kernel.

pub fn run( &self, stream: &Stream, _workspace: Workspace<'_>, args: CastArgs<'_, TIn, TOut>, ) -> Result<()>

Launch.