3 people 9af53a3c13 [Perf] Add tuned selective_state_update configs for H200 and RTX PRO … (#44251) ...

Signed-off-by: Majid Taheri Andani <tahemaji@amazon.com>
Co-authored-by: Majid Taheri Andani <tahemaji@amazon.com>
Co-authored-by: tomeras91 <57313761+tomeras91@users.noreply.github.com>

2026-06-02 23:59:01 -07:00

..

gdn

[BugFix] [GDN] Read linear_key_head_dim from hf_text_config for multimodal models (#43978)

2026-06-02 17:17:55 +03: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

linear_attn.py

[Kernel] Porting fuse_minimax_qk_norm to manual fusion (#43410)

2026-05-26 13:16:03 -07: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

[Bugfix ] fix bailing_moe_linear (#40859)

2026-04-27 22:39:23 -07:00

short_conv.py

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

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