Skip to content

Commit 16d4e2f

Browse files
authored
Remove memory_pool_limit in trt config (#7647)
Fixes Project-MONAI/model-zoo#568. ### Description Remove memory_pool_limit in trt config ### 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. - [ ] 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: YunLiu <[email protected]>
1 parent bff4b15 commit 16d4e2f

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

monai/networks/utils.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -840,7 +840,6 @@ def _onnx_trt_compile(
840840

841841
# set up the conversion configuration
842842
config = builder.create_builder_config()
843-
config.set_memory_pool_limit(trt.MemoryPoolType.WORKSPACE, 1 << 31)
844843
config.add_optimization_profile(profile)
845844
if precision == "fp16":
846845
config.set_flag(trt.BuilderFlag.FP16)

0 commit comments

Comments
 (0)