From 68c31cc559d6d2e3a0e9262f0f4cca0f7018eade Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 4 Mar 2022 12:31:59 +0100 Subject: [PATCH 1/3] Fix TRT `max_workspace_size` deprecation notice --- export.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/export.py b/export.py index 286df623d252..13fd31948183 100644 --- a/export.py +++ b/export.py @@ -217,7 +217,7 @@ def export_engine(model, im, file, train, half, simplify, workspace=4, verbose=F builder = trt.Builder(logger) config = builder.create_builder_config() - config.max_workspace_size = workspace * 1 << 30 + config.set_memory_pool_limit = workspace << 30 # max_workspace_size deprecated replacement flag = (1 << int(trt.NetworkDefinitionCreationFlag.EXPLICIT_BATCH)) network = builder.create_network(flag) From 9605b9564dab09c0939097fe80bc8a4accb20658 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 4 Mar 2022 12:45:30 +0100 Subject: [PATCH 2/3] Update export.py --- export.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/export.py b/export.py index 13fd31948183..a70b8fd95f32 100644 --- a/export.py +++ b/export.py @@ -217,7 +217,7 @@ def export_engine(model, im, file, train, half, simplify, workspace=4, verbose=F builder = trt.Builder(logger) config = builder.create_builder_config() - config.set_memory_pool_limit = workspace << 30 # max_workspace_size deprecated replacement + config.set_memory_pool_limit(workspace << 30) # max_workspace_size deprecated replacement flag = (1 << int(trt.NetworkDefinitionCreationFlag.EXPLICIT_BATCH)) network = builder.create_network(flag) From a59673f8a2cb3173d792808cad46797a06faea15 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 7 Mar 2022 13:52:02 +0100 Subject: [PATCH 3/3] Update export.py --- export.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/export.py b/export.py index 2d7d64ff9b6e..1e3d3e2f2e71 100644 --- a/export.py +++ b/export.py @@ -217,7 +217,8 @@ def export_engine(model, im, file, train, half, simplify, workspace=4, verbose=F builder = trt.Builder(logger) config = builder.create_builder_config() - config.set_memory_pool_limit(workspace << 30) # max_workspace_size deprecated replacement + config.max_workspace_size = workspace * 1 << 30 + # config.set_memory_pool_limit(trt.MemoryPoolType.WORKSPACE, workspace << 30) # fix TRT 8.4 deprecation notice flag = (1 << int(trt.NetworkDefinitionCreationFlag.EXPLICIT_BATCH)) network = builder.create_network(flag)