Artem Perevedentsev cb10b7e80b [GDN] Eliminate GPU->CPU sync in prepare_chunk_indices during prefill (#38361) ...

Signed-off-by: Artem Perevedentsev <aperevedents@nvidia.com>
Signed-off-by: Vadim Gimpelson <156319763+vadiklyutiy@users.noreply.github.com>

2026-04-03 13:38:02 +00:00

..

ops

[Kernel] Mamba support different layout for Conv state (#37416)

2026-04-03 01:50:09 +02: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

[Model][Spec Decode] Nemotron-H MTP and Mamba Speculative Decoding Support (#33726)

2026-02-24 09:49:56 -08:00

gdn_linear_attn.py

[GDN] Eliminate GPU->CPU sync in prepare_chunk_indices during prefill (#38361)

2026-04-03 13:38:02 +00:00

linear_attn.py

[V0 Deprecation] Refactor kv cache from list to element (#37487)

2026-03-23 20:10:11 -07:00

mamba_mixer2.py

[Kernel] Mamba support different layout for Conv state (#37416)

2026-04-03 01:50:09 +02:00

mamba_mixer.py

[Kernel] Mamba support different layout for Conv state (#37416)

2026-04-03 01:50:09 +02:00

mamba_utils.py

[Kernel] Mamba support different layout for Conv state (#37416)

2026-04-03 01:50:09 +02:00

short_conv.py

[Kernel] Mamba support different layout for Conv state (#37416)

2026-04-03 01:50:09 +02:00