Vedant V Jhaveri 25bd83033d Enable Piecewise CUDA Graph for NemotronH Hybrid (Mamba+Attention) Models (#19903)

2026-03-12 09:16:38 +08:00

..

cpu_graph_runner.py

Add torch.compile support for qwen3-next on CPU (#12444)

2026-02-26 23:28:03 -08:00

cuda_graph_runner.py

Feature/support longcat flash lite (#17838)

2026-03-09 23:00:11 +08:00

forward_batch_deepseek_mha_mixin.py

[2/n] deepseek_v2.py Refactor: Migrate MHA forward method in deepseek_v2.py (#16817)

2026-01-17 09:36:25 +08:00

forward_batch_info.py

Feature/support longcat flash lite (#17838)

2026-03-09 23:00:11 +08:00

hook_manager.py

Rename: --hooks to --forward-hooks (#13994)

2025-11-26 22:26:28 +08:00

input_buffers.py

[bugfix] disable share input buffer feature on npu due to accuracy issue (#19507)

2026-03-10 19:26:46 +08:00

mindspore_runner.py

[NPU]mindspore model support moe (#15363)

2026-02-02 17:52:49 +08:00

model_runner_kv_cache_mixin.py

[NPU] Bugfix for qwen35 on NPU (#19756)

2026-03-10 20:03:26 +08:00

model_runner.py

Enable Piecewise CUDA Graph for NemotronH Hybrid (Mamba+Attention) Models (#19903)

2026-03-12 09:16:38 +08:00

piecewise_cuda_graph_runner.py

Piecewise Cuda Graph set default (#16331)

2026-03-02 23:18:07 +08:00