Expand description
TRT JIT-OOM detection and single retry with halved workspace budget.
Constantsยง
- CHUNK_
CACHE_ ๐HIT_ THRESHOLD_ MS - Mirrors
trt_warmup::CACHE_HIT_THRESHOLD_MS. Used to classify a per-request inference as a probable TRT engine cache miss so the adaptive warmup task can proactively compile the engine.
Functionsยง
- embed_
with_ ๐trt_ retry - Wraps an embed call with the standard TRT JIT-OOM retry-once-with-halved-budget pattern.
- is_
trt_ ๐engine_ build_ fatal - Returns
truewhen an ORT error string indicates a TRT engine build failure severe enough that the worker should exit rather than retry. - is_
trt_ ๐jit_ oom - Returns
truewhen an ORT error string indicates a TRT JIT workspace overflow โ a condition that may resolve with a smaller batch or halved workspace budget.