vyre-libs 0.7.2

vyre Category A library ecosystem - pure-IR compositions over foundation IR and primitive-owned kernels
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

//! Linear-algebra sub-dialect: dot product, matmul, tiled matmul,
//! Strassen base case.
mod dot;
mod matmul;
mod matmul_strassen;
pub(crate) mod matmul_tiled;

pub use dot::{dot, Dot};
pub use matmul::{matmul, matmul_bias, Matmul, MatmulBias};
pub use matmul_strassen::{matmul_strassen_2x2, matmul_strassen_one_level};

// Keep the tiled builders on the linear-algebra sub-dialect surface.
pub use matmul_tiled::{matmul_bias_tiled, matmul_tiled, MatmulBiasTiled, MatmulTiled};
#[cfg(feature = "nn-linear-4bit")]
pub(crate) use matmul_tiled::{
    plan_matmul_kernel, F32MatmulMode, MatmulFallbackReason, MatmulKernelCapabilities,
    MatmulKernelPath, MatmulKernelPlan, MatrixShape,
};