MajidandMajid Taheri Andani c08ebebf30 [Perf] Add do_not_specialize to Mamba SSD chunk kernels (#43803) ...

Signed-off-by: Majid Taheri Andani <tahemaji@amazon.com>
Co-authored-by: Majid Taheri Andani <tahemaji@amazon.com>

2026-05-28 15:40:02 +00:00

..

gdn

[Perf][KDA] Fuse gate softplus, chunk-local cumsum, and RCP_LN2 scaling (#43667)

2026-05-28 13:47:08 +00:00

ops

[Perf] Add do_not_specialize to Mamba SSD chunk kernels (#43803)

2026-05-28 15:40:02 +00: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