Struct TailMaskPolicy 

pub struct TailMaskPolicy {
    pub original_count: u32,
    pub rounded_count: u32,
    pub tail_lanes: u32,
}

Result of coercing a logical element count up to the next power of two.

Backends that opt into the N6 substrate dispatch over rounded_count lanes (so every workgroup is uniform-shape, no boundary divergence on the last workgroup) and have the kernel guard each store with the tail-mask predicate lane_id < original_count. Threads beyond the original count no-op their stores.

The win is on tail handling for attention/softmax/reduce shapes where the workload is not a multiple of the workgroup size - without coercion the last workgroup runs with masked-out lanes that still incur scheduling cost; with coercion every workgroup is identical and the masked-out lanes are skipped via the predicate.

Fields

original_count: u32

Logical element count requested by the caller.

rounded_count: u32

Element count after rounding up to the next power of two. Equal to original_count when it is already a power of two.

tail_lanes: u32

Convenience: rounded_count - original_count. Lanes in this suffix range must be predicated off by the kernel.

Implementations

impl TailMaskPolicy

pub fn is_aligned(&self) -> bool

True when no rounding was needed; the dispatch can run as-is without a tail-mask predicate.

Trait Implementations

impl Clone for TailMaskPolicy

fn clone(&self) -> TailMaskPolicy

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Copy for TailMaskPolicy

impl Debug for TailMaskPolicy

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Eq for TailMaskPolicy

impl PartialEq for TailMaskPolicy

fn eq(&self, other: &TailMaskPolicy) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl StructuralPartialEq for TailMaskPolicy