Trait microgemm::Kernel

pub trait Kernel
where
    Self::Scalar: Copy + Zero + One,
{
    type Scalar;
    type Mr: ArrayLength + Multiply<Self::Nr>;
    type Nr: ArrayLength;

    const MR: usize = <Self::Mr>::USIZE;
    const NR: usize = <Self::Nr>::USIZE;

    // Required method
    fn microkernel(
        &self,
        alpha: Self::Scalar,
        lhs: &MatRef<'_, Self::Scalar>,
        rhs: &MatRef<'_, Self::Scalar>,
        beta: Self::Scalar,
        dst: &mut MatMut<'_, Self::Scalar>
    );

    // Provided methods
    fn gemm(
        &self,
        alpha: Self::Scalar,
        a: &MatRef<'_, Self::Scalar>,
        b: &MatRef<'_, Self::Scalar>,
        beta: Self::Scalar,
        c: &mut MatMut<'_, Self::Scalar>,
        pack_sizes: impl AsRef<PackSizes>,
        packing_buf: &mut [Self::Scalar]
    ) { ... }
    fn mr(&self) -> usize { ... }
    fn nr(&self) -> usize { ... }
}

Required Associated Types

type Scalar

type Mr: ArrayLength + Multiply<Self::Nr>

type Nr: ArrayLength

Provided Associated Constants

const MR: usize = <Self::Mr>::USIZE

const NR: usize = <Self::Nr>::USIZE

Required Methods

fn microkernel( &self, alpha: Self::Scalar, lhs: &MatRef<'_, Self::Scalar>, rhs: &MatRef<'_, Self::Scalar>, beta: Self::Scalar, dst: &mut MatMut<'_, Self::Scalar> )

Provided Methods

fn gemm( &self, alpha: Self::Scalar, a: &MatRef<'_, Self::Scalar>, b: &MatRef<'_, Self::Scalar>, beta: Self::Scalar, c: &mut MatMut<'_, Self::Scalar>, pack_sizes: impl AsRef<PackSizes>, packing_buf: &mut [Self::Scalar] )

fn mr(&self) -> usize

fn nr(&self) -> usize

Object Safety

This trait is not object safe.

Implementors

impl<T> Kernel for GenericKernel2x2<T>

where T: Copy + Zero + One + Add<Output = T> + Mul<Output = T>,

type Scalar = T

type Mr = UInt<UInt<UTerm, B1>, B0>

type Nr = UInt<UInt<UTerm, B1>, B0>

impl<T> Kernel for GenericKernel4x4<T>

where T: Copy + Zero + One + Add<Output = T> + Mul<Output = T>,

type Scalar = T

type Mr = UInt<UInt<UInt<UTerm, B1>, B0>, B0>

type Nr = UInt<UInt<UInt<UTerm, B1>, B0>, B0>

impl<T> Kernel for GenericKernel8x8<T>

where T: Copy + Zero + One + Add<Output = T> + Mul<Output = T>,

type Scalar = T

type Mr = UInt<UInt<UInt<UInt<UTerm, B1>, B0>, B0>, B0>

type Nr = UInt<UInt<UInt<UInt<UTerm, B1>, B0>, B0>, B0>

impl<T> Kernel for GenericKernel16x16<T>

where T: Copy + Zero + One + Add<Output = T> + Mul<Output = T>,

type Scalar = T

type Mr = UInt<UInt<UInt<UInt<UInt<UTerm, B1>, B0>, B0>, B0>, B0>

type Nr = UInt<UInt<UInt<UInt<UInt<UTerm, B1>, B0>, B0>, B0>, B0>

impl<T> Kernel for GenericKernel32x32<T>

where T: Copy + Zero + One + Add<Output = T> + Mul<Output = T>,

type Scalar = T

type Mr = UInt<UInt<UInt<UInt<UInt<UInt<UTerm, B1>, B0>, B0>, B0>, B0>, B0>

type Nr = UInt<UInt<UInt<UInt<UInt<UInt<UTerm, B1>, B0>, B0>, B0>, B0>, B0>