Commit 4e16400
fix: add torch.inference_mode() to parallel worker threads
torch.inference_mode() is thread-local — ThreadPoolExecutor workers
don't inherit the parent's context. Without it, parallel workers run
with autograd enabled (extra memory, different semantics than
sequential path).
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Signed-off-by: Sungsoo Ha <sungsooh@nvidia.com>1 parent f412e29 commit 4e16400
1 file changed
+1
-1
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
114 | 114 | | |
115 | 115 | | |
116 | 116 | | |
117 | | - | |
| 117 | + | |
118 | 118 | | |
119 | 119 | | |
120 | 120 | | |
| |||
0 commit comments