From ab8fa80fba1308aabae2465fa8271d7087ebb07e Mon Sep 17 00:00:00 2001 From: 16337 <1633794139@qq.com> Date: Thu, 29 Jan 2026 18:54:26 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E7=A7=BB=E9=99=A4=E5=B7=B2=E5=BA=9F?= =?UTF-8?q?=E5=BC=83=E7=9A=84=20BENCHMARK=20flag?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - BENCHMARK flag 在 TRT 8.5+ 已废弃 - tactic profiling 逻辑内部自动完成 --- build_engine.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/build_engine.py b/build_engine.py index fe2a8d8..f9ce340 100644 --- a/build_engine.py +++ b/build_engine.py @@ -263,8 +263,8 @@ def build_engine( if tactic_value > 0: config.set_tactic_sources(tactic_value) - if best: - config.set_flag(trt.BuilderFlag.BENCHMARK) + # BENCHMARK flag 在 TRT 8.5+ 已废弃,tactic profiling 逻辑内部自动完成 + # config.set_flag(trt.BuilderFlag.BENCHMARK) config.set_memory_pool_limit(trt.MemoryPoolType.WORKSPACE, workspace_mb * 1024 * 1024)