RishiAstra 46ae7f6666 [Bugfix] Mamba2 SSD varlen bug fix initstates decay, improve test, assert chunk pwr 2 (#21783) ...

Signed-off-by: Rishi Astra <40644327+RishiAstra@users.noreply.github.com>

2025-08-12 02:04:37 -07:00

..

test_causal_conv1d.py

…

test_mamba_mixer2.py

…

test_mamba_ssm_ssd.py

[Bugfix] Mamba2 SSD varlen bug fix initstates decay, improve test, assert chunk pwr 2 (#21783)

2025-08-12 02:04:37 -07:00

test_mamba_ssm.py

[Model] Mamba2 preallocate SSM output tensor to avoid d2d copy overhead (#21075)

2025-08-02 01:59:34 -07:00