Struct AvailableLineSizes

pub struct AvailableLineSizes {
    pub lhs: Vec<u8>,
    pub rhs: Vec<u8>,
    pub out: Vec<u8>,
}

Candidate line sizes supported for each tensor.

These lists begin with compiler-supported sizes and are progressively filtered based on problem shape divisibility and hardware constraints.

Fields

lhs: Vec<u8>

rhs: Vec<u8>

out: Vec<u8>

Implementations

impl AvailableLineSizes

pub fn from_elem_types<R: Runtime>(elem_in: &Elem, elem_out: &Elem) -> Self

pub fn filter_lhs_with_tensor( self, strides: &[usize], shape: &[usize], layout: MatrixLayout, ) -> Self

Filter available line sizes considering tensor shapes and strides for Lhs

pub fn filter_rhs_with_tensor( self, strides: &[usize], shape: &[usize], layout: MatrixLayout, ) -> Self

Filter available line sizes considering tensor shapes and strides for Rhs

pub fn filter_out_with_tensor(self, strides: &[usize], shape: &[usize]) -> Self

Filter available line sizes considering tensor shapes and strides for output

pub fn filter_lhs<F>(self, pred: F) -> Self

where F: FnMut(&u8) -> bool,

Filter available line sizes for Lhs

pub fn filter_rhs<F>(self, pred: F) -> Self

where F: FnMut(&u8) -> bool,

Filter available line sizes for Rhs

pub fn filter_out<F>(self, pred: F) -> Self

where F: FnMut(&u8) -> bool,

Filter available line sizes for output

pub fn pick_max(self) -> Result<MatmulLineSizes, MatmulSetupError>

Pick the largest remaining line size for each tensor

Trait Implementations

impl Clone for AvailableLineSizes

fn clone(&self) -> AvailableLineSizes

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for AvailableLineSizes

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

Formats the value using the given formatter.