Skip to main content

Module linear

Module linear 

Source
Expand description

Linear projection kernel.

Matches linear-projection-v1.yaml. y = xW^T + b — matrix multiply with optional bias.

Each function provides one of three backends:

  • fn linear_scalar(...) – Pure Rust scalar reference (ground truth)
  • unsafe fn linear_avx2(...) – AVX2 SIMD implementation
  • fn linear_ptx() -> &'static str – PTX assembly source string

Functions§

linear_avx2
AVX2 linear projection – delegates to scalar.
linear_ptx
PTX assembly for linear projection.
linear_scalar
Linear projection (scalar reference).