sungsoo haandThomas Parnell 4bf699d310 [Kernel] Support DS Mamba tail copy for MTP align mode (#45473) ...

Signed-off-by: Sungsoo Ha <sungsooh@nvidia.com>
Co-authored-by: Thomas Parnell <tom.parnell@gmail.com>

2026-06-16 22:50:30 +00:00

..

gdn

[PERF] [Qwen3.5] Split mixed prefill+decode batches: route decodes to the recurrent kernel (#44700)

2026-06-06 22:13:50 +08:00

linear

[Attention] Mamba attention module refactor - LINEAR (#43556)

2026-06-04 18:45:29 +08:00

ops

[Perf] Add tuned selective_state_update configs for H200 and RTX PRO … (#44251)

2026-06-02 23:59:01 -07: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

[Misc] Replace mamba_type string literals with MambaAttentionBackendEnum (#41188)

2026-05-11 03:59:36 +00:00

mamba_mixer2.py

[Bugfix][Hybrid][NemotronH] Fix mamba_cache_mode=all + speculative decoding crash (#41233)

2026-05-18 14:54:00 +03:00

mamba_mixer.py

[Perf][1/n] Eliminate various GPU<->CPU syncs (#41429)

2026-05-11 20:36:03 +00:00

mamba_utils.py

[Kernel] Support DS Mamba tail copy for MTP align mode (#45473)

2026-06-16 22:50:30 +00:00

short_conv.py

[Misc] Replace mamba_type string literals with MambaAttentionBackendEnum (#41188)

2026-05-11 03:59:36 +00:00