Struct WherePlan 

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

where(cond, a, b) plan with heterogeneous-dtype inputs.

T: Element is the value dtype (a / b / y). The cond is always u8. const N: usize is the tensor rank.

Implementations

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

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

Pick a kernel for desc. Returns Error::Unsupported if the value dtype isn’t wired today.

pub fn can_implement(&self, args: &WhereArgs<'_, T, N>) -> Result<()>

Validate that this plan can launch with args.

pub fn workspace_size(&self) -> usize

Workspace size in bytes. Always 0 for the trailblazer.

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: WhereArgs<'_, T, N>, ) -> Result<()>

Launch.