Struct ColMajorSpanMatmul

pub struct ColMajorSpanMatmul {}

Iterates on global matmuls in a col major fashion

Trait Implementations

impl CubeType for ColMajorSpanMatmul

type ExpandType = ColMajorSpanMatmulExpand

fn init(scope: &mut Scope, expand: Self::ExpandType) -> Self::ExpandType

Wrapper around the init method, necessary to type inference.

impl SpanMatmul for ColMajorSpanMatmul

fn execute<MP: MatmulPrecision, GMM: GlobalMatmul<MP>>( lhs: VirtualTensor<MP::EI>, rhs: VirtualTensor<MP::EI>, out: VirtualTensor<MP::EO, ReadWrite>, span: Span, acc: GMM::Accumulator, k_range: (u32, u32), quantization: CubeOption<Quantization<MP>>, config: GMM::Config, )

fn __expand_execute<MP: MatmulPrecision, GMM: GlobalMatmul<MP>>( 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, span: <Span as CubeType>::ExpandType, acc: <GMM::Accumulator as CubeType>::ExpandType, k_range: <(u32, u32) as CubeType>::ExpandType, quantization: <CubeOption<Quantization<MP>> as CubeType>::ExpandType, config: GMM::Config, ) -> <() as CubeType>::ExpandType