fix(autobatching): detach states to stop UMA autograd-graph leak (swap loop + chunked init) - #590

Merged

CompRhys merged 7 commits into

TorchSim:mainfrom

niklashoelter:fix/uma-inflight-energy-graph-leak

Jul 21, 2026

Commits on Jul 4, 2026

fix(autobatching): detach completed states to stop UMA forward-graph memory leak
niklashoelter
committed
24d7a13
test(autobatching): cover _detach_state_graph grad-drop for the UMA leak fix
niklashoelter
committed
de80139

Commits on Jul 5, 2026

fix(autobatching): detach states in _chunked_apply init pass
niklashoelter
committed
65c4af4
refactor(autobatching): promote detach_state_graph to public API
niklashoelter
committed
677ccef

Commits on Jul 21, 2026

Merge remote-tracking branch 'origin/main' into fix/uma-inflight-energy-graph-leak
CompRhys
committed
288e49e
maint: move state function to state.py
CompRhys
committed
6dc77ba
lint
CompRhys
committed
54bb565