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)