lnigam 1692f9e50b ggml : recurrent state rollback for ggml_ssm_scan (#26623) ...

* Initial changes for Recurrent state rollback for nemotron for cpu and cuda

* Removing CPU RS rollback. Will enable it in subsequent PRs

* addition of test case

* Removing assert and calling runtime API to check if op is supported

* removing extra API and updating the call sites for K

* replace static cuda detection to runtime fused_op api

* address review comments and fallback when SSM rollback not supprted

* Adding changes for supporting RS-rollback in CPU. Also added test-backend-ops for cpu and cuda

* removing memory manipulation as rs rollback is now supported in CPU

* removing the static probe which is not needed now

* correcting the format

* address review comments

* enabling test for all the backends, unsupported backends will fallback to CPU

* Apply suggestions from code review

Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>

* choose different graph based on the result of fused_ssm_op is supported or not and also handled memory->n_rs_seq >1 case incase of op is not supported

* Support K > 1 in ssm_scan for all backends

* Fix CI Issues

---------

Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
Co-authored-by: Gaurav Garg <gaugarg@nvidia.com>

2026-08-14 17:20:40 +03:00

..

afmoe.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

apertus.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

arcee.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

arctic.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

arwkv7.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

baichuan.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

bailingmoe2.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

bailingmoe.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

bert.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

bitnet.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

bloom.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

chameleon.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

chatglm.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

clip.cpp

llama: Restore quantization of mmprojs (#26818)

2026-08-10 11:58:32 +02:00

codeshell.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

cogvlm.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

cohere2.cpp

Add arch support for cohere2-MoE (#24260)

2026-06-13 19:49:00 +02:00

cohere2moe.cpp

llama : load MTP tensors only if they are really used (#26296)

2026-07-31 14:57:02 +02:00

command-r.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

dbrx.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

deci.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

deepseek2.cpp

model : support MTP in GLM-4.7-Flash (#24868)

2026-08-03 20:27:52 +02:00

deepseek2ocr.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

deepseek4.cpp

model : allow reshape of tensors during load (#26531)

2026-08-04 09:06:44 +03:00

deepseek32.cpp

llama : MTP support for DeepSeek V3.2 (#26457)

2026-08-03 08:25:01 +02:00

deepseek.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

delta-net-base.cpp

llama: refactor fused ops (#24646)

2026-07-08 18:18:09 +08:00

dflash.cpp

dflash : clarify output logging of target_layer_ids (#27013)

2026-08-14 06:57:05 +02:00

dots1.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

dream.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

eagle3.cpp

spec: add eagle3-v3 support for gpt-oss model (#25794)

2026-07-28 09:58:16 +03:00

ernie4-5-moe.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

ernie4-5.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

eurobert.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

exaone4.cpp

model : fix SWA not being enabled for EXAONE 4.5 (#26848)

2026-08-11 07:20:17 +03:00

exaone-moe.cpp

minor : fix lint issues (#24165)

2026-06-05 11:17:54 +03:00

exaone.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

falcon-h1.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

falcon.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

gemma2.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

gemma3.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

gemma3n.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

gemma4-assistant.cpp

mtp: support for gemma-4 E2B and E4B assistants (#24282)

2026-06-08 13:48:52 -07:00

gemma4.cpp

llama: move suppress_tokens handling to common/sampling (#26276)

2026-07-29 18:02:30 +02:00

gemma-embedding.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

gemma.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

glm4-moe.cpp

minor : fix lint issues (#24165)

2026-06-05 11:17:54 +03:00

glm4.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

glm-dsa.cpp

llama : load MTP tensors only if they are really used (#26296)

2026-07-31 14:57:02 +02:00

gpt2.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

gptneox.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

granite-hybrid.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

granite-moe.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

granite-switch.cpp

model : Granite-Switch Architecture (#25107)

2026-08-10 09:53:46 +02:00

granite.cpp

model, mtmd: Granite4 Vision (#23545)

2026-06-05 17:44:59 +02:00

grok.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

grovemoe.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

hunyuan-dense.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

hunyuan-moe.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

hunyuan-vl.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

hy-v3.cpp

llama : load MTP tensors only if they are really used (#26296)

2026-07-31 14:57:02 +02:00

internlm2.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

jais2.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

jais.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

jamba.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

jina-bert-v2.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

jina-bert-v3.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

kimi-linear.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

laguna.cpp

model: Add Laguna-S-2.1 LLM_TYPE (#26233)

2026-07-28 21:02:33 +02:00

lfm2.cpp

model : Add label for LFM2.5-230M (#25008)

2026-06-25 18:58:52 +02:00

lfm2moe.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

llada-moe.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

llada.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

llama4.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

llama-embed.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

llama.cpp

spec: add EAGLE3 speculative decoding support (#18039)

2026-06-12 10:21:06 +03:00

maincoder.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

mamba2.cpp

mamba2: remove hardcoded 2x expansion factor and invalid d_inner % d_state check (#23082)

2026-06-26 08:50:54 +03:00

mamba-base.cpp

ggml : recurrent state rollback for ggml_ssm_scan (#26623)

2026-08-14 17:20:40 +03:00

mamba.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

mellum.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

mimo2.cpp

model : load MiMo V2 MTP tensors only if used (#26412)

2026-08-02 09:03:05 +02:00

minicpm3.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

minicpm.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

minimax-m2.cpp

spec: add Minimax2 eagle3 support

2026-07-13 15:22:37 +02:00

minimax-m3.cpp

graph : fix unused input tensors in minimax m3 graph (#26519)

2026-08-03 17:32:01 +03:00

mistral3.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

mistral4.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

models.h

mtmd: support pocket-tts (#26871)

2026-08-11 14:18:30 +02:00

modern-bert.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

mpt.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

muse-glimmer.cpp

model: Muse Glimmer Support (#26841)

2026-08-10 13:07:27 +02:00

nanbeige.cpp

model: Add support for Nanbeige4.2 (#25994)

2026-07-27 17:04:18 +02:00

nemotron-h-moe.cpp

model: add MTP support for Nemotron model (#26725)

2026-08-10 11:25:24 +03:00

nemotron-h.cpp

Dflash support for nemotron-3.5 (#26905)

2026-08-11 18:46:26 +05:30

nemotron.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

neo-bert.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

nomic-bert-moe.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

nomic-bert.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

olmo2.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

olmo.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

olmoe.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

openai-moe.cpp

spec: add eagle3-v3 support for gpt-oss model (#25794)

2026-07-28 09:58:16 +03:00

openelm.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

orion.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

paddleocr.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

pangu-embed.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

phi2.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

phi3.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

phimoe.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

plamo2.cpp

ggml : recurrent state rollback for ggml_ssm_scan (#26623)

2026-08-14 17:20:40 +03:00

plamo3.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

plamo.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

plm.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

pockettts.cpp

mtmd: support pocket-tts (#26871)

2026-08-11 14:18:30 +02:00

qwen2.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

qwen2moe.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

qwen2vl.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

qwen3.cpp

spec: add EAGLE3 speculative decoding support (#18039)

2026-06-12 10:21:06 +03:00

qwen3moe.cpp

spec: add EAGLE3 speculative decoding support (#18039)

2026-06-12 10:21:06 +03:00

qwen3next.cpp

model: MTP support for Qwen3-Next (#25589)

2026-08-03 17:15:01 +09:00

qwen3tts.cpp

mtmd: support Qwen3-TTS (note: breaking change to llama-tts binary) (#26254)

2026-08-04 17:26:15 +02:00

qwen3vl.cpp

mtmd: support Qwen3-TTS (note: breaking change to llama-tts binary) (#26254)

2026-08-04 17:26:15 +02:00

qwen3vlmoe.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

qwen35.cpp

llama : load MTP tensors only if they are really used (#26296)

2026-07-31 14:57:02 +02:00

qwen35moe.cpp

llama : load MTP tensors only if they are really used (#26296)

2026-07-31 14:57:02 +02:00

qwen.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

refact.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

rnd1.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

rwkv6-base.cpp

models : deduplicate delta-net graphs for Qwen family (#19597)

2026-02-16 14:35:04 +02:00

rwkv6.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

rwkv6qwen2.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

rwkv7-base.cpp

models : deduplicate delta-net graphs for Qwen family (#19597)

2026-02-16 14:35:04 +02:00

rwkv7.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

seed-oss.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

smallthinker.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

smollm3.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

stablelm.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

starcoder2.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

starcoder.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

step35.cpp

llama : load MTP tensors only if they are really used (#26296)

2026-07-31 14:57:02 +02:00

t5.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

t5encoder.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

talkie.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00

wavtokenizer-dec.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

xverse.cpp

hparams : refactor hparams.n_layer (#24060)

2026-06-05 11:09:36 +03:00