XW-AIOT/security-ai-edge
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 移除已废弃的 BENCHMARK flag

Browse Source 
- BENCHMARK flag 在 TRT 8.5+ 已废弃
- tactic profiling 逻辑内部自动完成
This commit is contained in:
16337
2026-01-29 18:54:26 +08:00
parent 4103b503db
commit ab8fa80fba
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
build_engine.py
View File

@@ -263,8 +263,8 @@ def build_engine(
if tactic_value > 0: if tactic_value > 0:
config.set_tactic_sources(tactic_value) config.set_tactic_sources(tactic_value)
if best: # BENCHMARK flag 在 TRT 8.5+ 已废弃，tactic profiling 逻辑内部自动完成
config.set_flag(trt.BuilderFlag.BENCHMARK) # config.set_flag(trt.BuilderFlag.BENCHMARK)
config.set_memory_pool_limit(trt.MemoryPoolType.WORKSPACE, workspace_mb * 1024 * 1024) config.set_memory_pool_limit(trt.MemoryPoolType.WORKSPACE, workspace_mb * 1024 * 1024)