From 73f0e3ff6fa28ea406a225be2547a3cf8363b2dd Mon Sep 17 00:00:00 2001
From: binliu <binliu@nvidia.com>
Date: Thu, 25 Apr 2024 13:28:31 +0000
Subject: [PATCH] update the trt export ir parameter

Signed-off-by: binliu <binliu@nvidia.com>
---
 monai/networks/utils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/monai/networks/utils.py b/monai/networks/utils.py
index 7e3d2e2170..152911f443 100644
--- a/monai/networks/utils.py
+++ b/monai/networks/utils.py
@@ -986,6 +986,7 @@ def scale_batch_size(input_shape: Sequence[int], scale_num: int):
                     inputs=input_placeholder,
                     enabled_precisions=convert_precision,
                     device=target_device,
+                    ir="torchscript",
                     **kwargs,
                 )