mamba: avoid redundant HBM reloads in causal_conv1d_update shift loop - #4460

Merged

wdykas merged 6 commits into

NVIDIA:mainfrom

wdykas:fix/causal-conv1d-shift-loop-hbm-reads

Apr 29, 2026

Commits on Apr 24, 2026

mamba: avoid redundant HBM reloads in causal_conv1d_update shift loop

authored and
root
committed
31fa060
lint

root
authored and
root
committed
e5c3fdc

Commits on Apr 28, 2026

ADD TEST
William Dykas
committed
2082ea7
Merge branch 'fix/causal-conv1d-shift-loop-hbm-reads' of github.com:wdykas/Megatron-LM into fix/causal-conv1d-shift-loop-hbm-reads
William Dykas
committed
48afcad
Merge branch 'main' into fix/causal-conv1d-shift-loop-hbm-reads
wdykas
authored
6d91c9b

Commits on Apr 29, 2026

Merge branch 'main' into fix/causal-conv1d-shift-loop-hbm-reads
wdykas
authored
f151db1