Expand description
State-Space Model (SSM) scan kernel.
Matches ssm-kernel-v1.yaml.
h_t = A_bar * h_{t-1} + B_bar * x_t
y_t = C * h_t
Functions§
- ssm_
scan_ ⚠avx2 - AVX2 implementation of the SSM sequential scan.
- ssm_
scan_ ptx - PTX assembly for the SSM scan kernel.
- ssm_
scan_ scalar - Scalar reference implementation of the SSM sequential scan.