Struct Gemm 

pub struct Gemm<'a> {
    pub x: &'a [f32],
    pub m: usize,
    pub k: usize,
    pub w: &'a [f32],
    pub n: usize,
    pub b: Option<&'a [f32]>,
    pub ep: Epilogue,
}

One GEMM with its epilogue, y = ep(x wᵀ + b), split into tiles that any thread may run. Every output element is computed the same way whatever the split, so the split is free to follow the thread count.

Fields

x: &'a [f32]

[m, k].

m: usize

Rows of x and y.

k: usize

The reduction length.

w: &'a [f32]

[n, k] as pack lays it out.

n: usize

Columns of y.

b: Option<&'a [f32]>

[n].

ep: Epilogue

What happens to each result.

Implementations

impl Gemm<'_>

pub fn run( &self, y: &mut [f32], threads: usize, spawn: impl FnOnce(usize, &(dyn Fn(usize, &mut [f32]) + Sync)), )

Runs the GEMM into y, handing spawn a task count and the task body to run for each. A task gets scratch_len floats of scratch of its own.

Panics

If a length does not match the shape.

Trait Implementations

impl<'a> Clone for Gemm<'a>

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<'a> Copy for Gemm<'a>

impl<'a> Debug for Gemm<'a>

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

Formats the value using the given formatter.