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
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

//! Linear layer: `y = x @ W + b`.
//!
//! Audit-fix A30 split this module by concern: `linear` builder + struct in
//! `mod.rs`/`builder.rs`, the tiled variants in `tiled.rs`, the fused
//! `linear_relu` in `relu.rs`, `rms_norm_linear` in `rms_norm.rs`, and tests
//! in `tests.rs`.

mod batch_matmul;
mod builder;
mod fused_activation;
#[cfg(feature = "nn-linear-4bit")]
mod linear_4bit;
mod relu;
mod rms_norm;
mod silu;
mod tiled;

pub use batch_matmul::batch_matmul;
pub use builder::{linear, Linear};
#[cfg(feature = "nn-linear-4bit")]
pub use linear_4bit::{
    linear_4bit, linear_4bit_affine_grouped, linear_4bit_affine_grouped_typed,
    QuantizedLinear4BitSpec,
};
pub use relu::linear_relu;
pub use rms_norm::{rms_norm_linear, try_rms_norm_linear};
pub use silu::linear_silu;
pub use tiled::{linear_tiled, linear_tiled_reference};

#[cfg(test)]
mod tests;