Struct OneToOneMatmul 

pub struct OneToOneMatmul<MP: MatmulPrecision, GMM: GlobalMatmul<MP>, C: CubeDispatch> { /* private fields */ }

Executes matrix multiplication at the batch level, assigning each cube to a single global matmul.

Note: This algorithm requires one cube per global matmul; insufficient cubes will result in incomplete computations.

Trait Implementations

impl<MP: MatmulPrecision, GMM: GlobalMatmul<MP>, C: CubeDispatch> BatchMatmul<MP> for OneToOneMatmul<MP, GMM, C>

type Config = Config<<GMM as GlobalMatmul<MP>>::Config, C>

fn execute( lhs: VirtualTensor<MP::EI>, rhs: VirtualTensor<MP::EI>, out: VirtualTensor<MP::EO, ReadWrite>, size_k: u32, quantization: CubeOption<Quantization<MP>>, config: Self::Config, )

Performs batchwise matrix multiplication over tensors.

fn __expand_execute( context: &mut Scope, lhs: <VirtualTensor<MP::EI> as CubeType>::ExpandType, rhs: <VirtualTensor<MP::EI> as CubeType>::ExpandType, out: <VirtualTensor<MP::EO, ReadWrite> as CubeType>::ExpandType, size_k: <u32 as CubeType>::ExpandType, quantization: <CubeOption<Quantization<MP>> as CubeType>::ExpandType, config: Self::Config, ) -> <() as CubeType>::ExpandType