fix: 移除已废弃的 BENCHMARK flag

- BENCHMARK flag 在 TRT 8.5+ 已废弃 - tactic profiling 逻辑内部自动完成
2026-01-29 18:54:26 +08:00
parent 4103b503db
commit ab8fa80fba
1 changed files with 2 additions and 2 deletions
--- 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)