vyre-libs 0.6.2

vyre Category A library ecosystem - pure-IR compositions over vyre-ops hardware primitives
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14

//! 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};

// H1 Strassen base case re-export at the math root for parity with
// existing dot/matmul exports.
pub use matmul_tiled::{matmul_bias_tiled, matmul_tiled, MatmulBiasTiled, MatmulTiled};