Skip to main content

Module simd

Module simd 

Source
Expand description

SIMD-accelerated math primitives for neural forward passes.

Provides AVX2-accelerated dot product, matrix-vector multiply, and activation functions with automatic runtime feature detection and scalar fallback.

On x86_64 targets with the std feature enabled, these functions use runtime is_x86_feature_detected!("avx2") to select AVX2 when available. On all other targets (or without std), they fall back to scalar implementations.

Re-exports§

pub use ops::simd_dot;
pub use ops::simd_exp;
pub use ops::simd_mat_vec;
pub use ops::simd_sigmoid;
pub use ops::simd_silu;
pub use ops::simd_tanh;

Modules§

ops
Core SIMD-accelerated operations: dot product, matrix-vector multiply, and activation functions.