Thomas ParnellandChih-Chieh-Yang 5e25b12236 [Kernel] [Mamba] Remove BLOCK_H=1 from list of tuneable configurations for _chunk_cumsum_fwd_kernel (#25197) ...

Signed-off-by: Thomas Parnell <tpa@zurich.ibm.com>
Co-authored-by: Chih-Chieh-Yang <chih.chieh.yang@ibm.com>

2025-09-23 23:23:30 +00:00

..

ops

[Kernel] [Mamba] Remove BLOCK_H=1 from list of tuneable configurations for _chunk_cumsum_fwd_kernel (#25197)

2025-09-23 23:23:30 +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

[V1] Remove V0 code paths for Hybrid models (#25400)

2025-09-23 08:26:13 -07:00

linear_attn.py

Remove redundant mutates_args and dispatch_key for direct_register_custom_op (#25512)

2025-09-23 22:48:40 +00:00

mamba_mixer2.py

Remove redundant mutates_args and dispatch_key for direct_register_custom_op (#25512)

2025-09-23 22:48:40 +00:00

mamba_mixer.py

Remove redundant mutates_args and dispatch_key for direct_register_custom_op (#25512)

2025-09-23 22:48:40 +00:00

mamba_utils.py

[V1] Remove V0 code paths for Hybrid models (#25400)

2025-09-23 08:26:13 -07:00

short_conv.py

Remove redundant mutates_args and dispatch_key for direct_register_custom_op (#25512)

2025-09-23 22:48:40 +00:00