vyre-libs 0.6.5

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

//! Linear-layer sub-dialect: affine transforms built on `math::linalg`.
mod inner;

pub use inner::{
    batch_matmul, linear, linear_relu, linear_silu, linear_tiled, linear_tiled_reference,
    rms_norm_linear, try_rms_norm_linear, Linear,
};
#[cfg(feature = "nn-linear-4bit")]
pub use inner::{
    linear_4bit, linear_4bit_affine_grouped, linear_4bit_affine_grouped_planner_evidence,
    linear_4bit_affine_grouped_typed, QuantizedLinear4BitPlannerEvidence, QuantizedLinear4BitSpec,
    LINEAR_4BIT_AFFINE_GROUPED_OUTPUT_DRIFT_ABS_TOLERANCE,
};