From bd9bc2e05dc4fcd6c536b4fdae7e136d1dd7fc69 Mon Sep 17 00:00:00 2001 From: Roman Meshko Date: Sat, 23 May 2026 22:07:01 +0300 Subject: [PATCH] Changed to use prepare for TensorRT --- scripts/jetson/generate_int8_cache.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/scripts/jetson/generate_int8_cache.py b/scripts/jetson/generate_int8_cache.py index 7d6d3d5..622df9e 100644 --- a/scripts/jetson/generate_int8_cache.py +++ b/scripts/jetson/generate_int8_cache.py @@ -98,6 +98,13 @@ def main(): print(f"Cache written → {args.output}") onnx_data = Path(args.onnx).read_bytes() + try: + from engines.onnx_tensorrt_compat import prepare_for_tensorrt + + onnx_data = prepare_for_tensorrt(onnx_data) + print("Prepared ONNX model for TensorRT static Jetson build") + except Exception as e: + print(f"WARNING: ONNX TensorRT compatibility preparation failed: {e}", file=sys.stderr) logger = trt.Logger(trt.Logger.INFO) explicit_batch = 1 << int(trt.NetworkDefinitionCreationFlag.EXPLICIT_BATCH)