vllm.model_executor.layers.mamba.ops ¶

Modules:

• causal_conv1d –
• cpu –
• gather_initial_states –
• gdn_chunk_cutedsl –
• mamba_ssm –
• replayssm_config –

  Launch-config selection for the ReplaySSM Mamba2 output_only decode kernel.

• selective_state_update_replayssm_output_only –
• ssd_bmm –
• ssd_combined –
• ssu_dispatch –

  Dispatch module for Mamba selective state update (SSU) backends.

• triton_helpers –