Struct MatmulProblem

pub struct MatmulProblem {
    pub m: usize,
    pub n: usize,
    pub k: usize,
    pub batches: (Vec<usize>, Vec<usize>),
    pub lhs_layout: MatrixLayout,
    pub rhs_layout: MatrixLayout,
    pub lhs_line_size: u8,
    pub rhs_line_size: u8,
    pub out_line_size: u8,
}

Description of a matmul problem to solve, regardless of actual data

Fields

m: usize

n: usize

k: usize

batches: (Vec<usize>, Vec<usize>)

lhs_layout: MatrixLayout

rhs_layout: MatrixLayout

lhs_line_size: u8

rhs_line_size: u8

out_line_size: u8

Implementations

impl MatmulProblem

pub fn check_config<B: BatchConfig>( &self, config: &B, ) -> Result<(), MatmulInvalidProblem>

Asserts that the problem can be solved with the given batch matmul configs

Panics:

• If dimensions of the problem are larger than allowed by the config
• If line sizes do not divide well the dimension in which they are aligned

Trait Implementations

impl Clone for MatmulProblem

fn clone(&self) -> MatmulProblem

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MatmulProblem

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

Formats the value using the given formatter.