Expand description
Allocator housekeeping for a process that runs ONNX Runtime or converts many documents — a server, an agent tool, a batch worker, a binding (#263, #669, #871).
A conversion churns hundreds of MB of short-lived buffers (page bitmaps, ONNX Runtime tensors) across many threads. glibc’s allocator keeps what they free: measured on the MCP server, 8 MB was live after a 9-page PDF while RSS sat at 0.5–1.2 GB — all of it free memory held in 21 per-thread arenas, growing call over call. Two causes, two calls:
pin_mmap_threshold, once at startup: glibc raises its mmap threshold (up to 32 MiB) every time a large mapped block is freed, after which blocks of that size are carved from the arenas, where they fragment and stay resident. A fixed threshold turns the adjustment off, so large buffers are mapped and unmapped again. docling-onnx pins it when it creates the ONNX Runtime environment, so every process that runs a model has it — the CLI and the bindings too, not only the servers (#871): the TableFormer decoder allocates a KV cache one token longer on every step, so a long table frees hundreds of ever-larger blocks inside a single conversion. With the threshold free to rise they all landed in the arenas, and on linux/arm64 (ONNX Runtime 1.31) few of the freed blocks were reused: 4.3 GB of free heap 608 steps into one table, an OOM under a 6 GiB limit that 2.22.0 converted in 3.3 GB.release_free_memory, after each conversion: hands the free pages still inside the arenas back to the OS.
Both are glibc-only (Linux with the GNU C library) and do nothing elsewhere — musl and the macOS/Windows allocators lack the pattern and the calls.
Structs§
- Trim
OnDrop release_free_memorywhen dropped: bound at the top of a conversion’s scope (let _trim = TrimOnDrop;), the trim runs however the scope ends — return, early?, panic unwinding — after the buffers declared in it are gone. The bindings put one at every entry point that converts (#871: Node’sPipeline.convertAsyncpath had no trim, and RSS between documents climbed 950 → 1,566 MiB over five PDFs).
Constants§
- MMAP_
THRESHOLD - The fixed mmap threshold
pin_mmap_thresholdsets. Any fixed value stops the growth; measured on a warm MCP server (9-page paper, 4 vCPU, ONNX arena off) RSS held at ~475 MB with 1, 4 or 16 MiB against 500 → 750 MB without, and from 4 MiB up the conversions were as fast as glibc’s default — at 128 KiB every ONNX intermediate tensor was mapped and conversions ran ~8% slower.
Functions§
- pin_
mmap_ threshold - Pin glibc’s mmap threshold at
MMAP_THRESHOLDfor the rest of the process, unless the operator chose one (GLIBC_TUNABLESnamingglibc.malloc.mmap_threshold, orMALLOC_MMAP_THRESHOLD_). Idempotent; call it early — blocks the arenas already hold stay there. - release_
free_ memory - Return the free memory glibc still holds in its arenas to the OS — after a conversion, once its buffers are dropped.