Struct BinaryCmpPlan 

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

Binary comparison plan.

T: Element is the input element type (today: must be f32). Output is always u8. const N: usize is the tensor rank.

Implementations

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

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

Pick a kernel for desc. Returns Error::Unsupported if the (kind, T::KIND) pair isn’t wired today.

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

Validate that this plan can launch with args.

Accepts contig and strided / broadcast operands. Same broadcast rules as crate::BinaryPlan::can_implement.

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

Launch.