Struct AvailableVectorSizes 

pub struct AvailableVectorSizes {
    pub lhs: Vec<VectorSize>,
    pub rhs: Vec<VectorSize>,
    pub out: Vec<VectorSize>,
}

Candidate vector 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<VectorSize>

rhs: Vec<VectorSize>

out: Vec<VectorSize>

Implementations

impl AvailableVectorSizes

pub fn from_type_size_tma<R: Runtime>( client: &ComputeClient<R>, elem_out: usize, ) -> Self

pub fn from_type_sizes<R: Runtime>( client: &ComputeClient<R>, elem_lhs: usize, elem_rhs: usize, elem_out: usize, ) -> Self

pub fn filter_lhs_with_tensor( self, strides: &Strides, shape: &Shape, layout: MatrixLayout, ) -> Self

Filter available vector sizes considering tensor shapes and strides for Lhs

pub fn filter_rhs_with_tensor( self, strides: &Strides, shape: &Shape, layout: MatrixLayout, ) -> Self

Filter available vector sizes considering tensor shapes and strides for Rhs

pub fn filter_out_with_tensor(self, strides: &Strides, shape: &Shape) -> Self

Filter available vector sizes considering tensor shapes and strides for output

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

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

Filter available vector sizes for Lhs

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

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

Filter available vector sizes for Rhs

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

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

Filter available vector sizes for output

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

Pick the largest remaining vector size for each tensor

Trait Implementations

impl Clone for AvailableVectorSizes

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for AvailableVectorSizes

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

Formats the value using the given formatter.