File tree Expand file tree Collapse file tree 3 files changed +23
-2
lines changed
Expand file tree Collapse file tree 3 files changed +23
-2
lines changed Original file line number Diff line number Diff line change 11# CHANGELOG
22
33
4+ ## v0.20.5 (2026-03-09)
5+
6+ ### Bug Fixes
7+
8+ - ** vis** : Prevent OOM kill on 1M-node ELK render
9+ ([ #128 ] ( https://github.com/johnmarktaylor91/torchlens/pull/128 ) ,
10+ [ ` d9a1525 ` ] ( https://github.com/johnmarktaylor91/torchlens/commit/d9a15259e452bf3c224732a0bc5a1671f9cbb56e ) )
11+
12+ The 1M-node render was OOM-killed at ~ 74GB RSS because: 1. Model params (~ 8-10GB) stayed alive
13+ during ELK subprocess 2. preexec_fn forced fork+exec, COW-doubling the 74GB process 3. Heap/stack
14+ formulas produced absurd values (5.6TB heap, 15GB stack) 4. No memory cleanup before subprocess
15+ launch
16+
17+ Changes: - render_large_graph.py: separate log_forward_pass from render_graph, free model/autograd
18+ before ELK render - elk_layout.py: cap heap at 64GB, stack floor 4096MB/cap 8192MB, write JSON to
19+ temp file (free string before subprocess), gc.collect before subprocess, set RLIMIT_STACK at
20+ module level (removes preexec_fn and the forced fork+exec)
21+
22+ Co-Authored-By: Claude Opus 4.6 < noreply@anthropic.com >
23+
24+
425## v0.20.4 (2026-03-09)
526
627### Bug Fixes
Original file line number Diff line number Diff line change @@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
44
55[project ]
66name = " torchlens"
7- version = " 0.20.4 "
7+ version = " 0.20.5 "
88description = " A package for extracting activations from PyTorch models"
99readme = " README.md"
1010license = " GPL-3.0-only"
Original file line number Diff line number Diff line change 99 first imported. See the bottom of this file for details.
1010"""
1111
12- __version__ = "0.20.4 "
12+ __version__ = "0.20.5 "
1313
1414# ---- Public API: user-facing entry points --------------------------------
1515
You can’t perform that action at this time.
0 commit comments