From 16d4e2fcdbc6a7be5bf9df02fc12f34eee5f26b9 Mon Sep 17 00:00:00 2001 From: YunLiu <55491388+KumoLiu@users.noreply.github.com> Date: Tue, 16 Apr 2024 22:05:35 +0800 Subject: [PATCH] Remove memory_pool_limit in trt config (#7647) Fixes https://github.com/Project-MONAI/model-zoo/issues/568. ### Description Remove memory_pool_limit in trt config ### Types of changes - [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 <55491388+KumoLiu@users.noreply.github.com> --- monai/networks/utils.py | 1 - 1 file changed, 1 deletion(-) diff --git a/monai/networks/utils.py b/monai/networks/utils.py index 4e6699f16b..7e3d2e2170 100644 --- a/monai/networks/utils.py +++ b/monai/networks/utils.py @@ -840,7 +840,6 @@ def _onnx_trt_compile( # set up the conversion configuration config = builder.create_builder_config() - config.set_memory_pool_limit(trt.MemoryPoolType.WORKSPACE, 1 << 31) config.add_optimization_profile(profile) if precision == "fp16": config.set_flag(trt.BuilderFlag.FP16)