From 6a130cc7f659efe66cf651369e35d4a93b2ec28a Mon Sep 17 00:00:00 2001
From: binliunls <107988372+binliunls@users.noreply.github.com>
Date: Fri, 26 Apr 2024 13:18:06 +0800
Subject: [PATCH] 7713 Update TRT parameter (#7714)

Fixes #7713  .

### Description

A few sentences describing the changes proposed in this pull request.

### Types of changes
<!--- Put an `x` in all the boxes that apply, and remove the not
applicable items -->
- [x] Non-breaking change (fix or new feature that would not break
existing functionality).
- [ ] Breaking change (fix or new feature that would cause existing
functionality to change).
- [ ] New tests added to cover the changes.
- [x] Integration tests passed locally by running `./runtests.sh -f -u
--net --coverage`.
- [ ] Quick tests passed locally by running `./runtests.sh --quick
--unittests --disttests`.
- [ ] In-line docstrings updated.
- [ ] Documentation updated, tested `make html` command in the `docs/`
folder.

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,
                 )