Trait 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: PackSizes,
        packing_buf: &mut [Self::Scalar],
    ) { ... }
    fn mr(&self) -> usize { ... }
    fn nr(&self) -> usize { ... }
}

Provided Associated Constants

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

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

Required Associated Types

type Scalar

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

type Nr: ArrayLength

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: PackSizes, packing_buf: &mut [Self::Scalar], )

fn mr(&self) -> usize

fn nr(&self) -> usize

Dyn Compatibility

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so 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>