Skip to content

Commit 9b65063

Browse files
author
semantic-release
committed
chore(release): 0.20.5
1 parent ef80cfe commit 9b65063

File tree

3 files changed

+23
-2
lines changed

3 files changed

+23
-2
lines changed

CHANGELOG.md

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,27 @@
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

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
44

55
[project]
66
name = "torchlens"
7-
version = "0.20.4"
7+
version = "0.20.5"
88
description = "A package for extracting activations from PyTorch models"
99
readme = "README.md"
1010
license = "GPL-3.0-only"

torchlens/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
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

0 commit comments

Comments
 (0)