Skip to content

fix: deterministic ORT session release (int8 OOM) - #149

Merged
ronaldtse merged 1 commit into
mainfrom
fix/ort-release
Sep 3, 2026
Merged

fix: deterministic ORT session release (int8 OOM)#149
ronaldtse merged 1 commit into
mainfrom
fix/ort-release

Conversation

@ronaldtse

Copy link
Copy Markdown
Contributor

The 2.1 parity crashed ~3min into the int8 stage twice; the zip itself decodes clean locally (20/20 pairs). Root cause: cumulative ORT arenas across the interleaved precision stages. Sessions now release in a finally block; parity tests green.

Six interleaved session lifetimes (3 precisions x parity+margins)
plus the torch reference OOMed the 32GB container at the int8 stage;
del + gc in a finally block instead of waiting on GC.
@ronaldtse
ronaldtse merged commit 86cb354 into main Sep 3, 2026
11 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant