daniserebandRobert Shaw f4a0921c9c [Performance] Tune Mamba selective scan kernel for B200 (#32873) ...

Signed-off-by: Daniel Serebrenik <daserebrenik@nvidia.com>
Co-authored-by: Robert Shaw <114415538+robertgshaw2-redhat@users.noreply.github.com>

2026-01-26 05:56:54 -08:00

..

ops

[Performance] Tune Mamba selective scan kernel for B200 (#32873)

2026-01-26 05:56:54 -08:00

__init__.py

[Kernel/Model] Migrate mamba_ssm and causal_conv1d kernels to vLLM (#7651)

2024-08-28 15:06:52 -07:00

abstract.py

[V1][Hybrid] Mamba Prefix Caching with align mode (#30877)

2026-01-23 09:56:48 -08:00

linear_attn.py

[1/N][Attention] Restructure attention: move files (#31916)

2026-01-09 13:10:24 -08:00

mamba_mixer2.py

[Performance] Tune Mamba selective scan kernel for B200 (#32873)

2026-01-26 05:56:54 -08:00

mamba_mixer.py

[V1][Hybrid] Mamba Prefix Caching with align mode (#30877)

2026-01-23 09:56:48 -08:00

mamba_utils.py

[V1][Hybrid] Mamba Prefix Caching with align mode (#30877)

2026-01-23 09:56:48 -08:00

short_conv.py

[1/N][Attention] Restructure attention: move files (#31916)

2026-01-09 13:10:24 -08:00