未验证 提交 236201c2 编写于 作者: Z Zhaolong Xing 提交者: GitHub

Merge pull request #15350 from NHZlX/fix_bug_for_precditor

fix analysis config bug
...@@ -127,6 +127,7 @@ void contrib::AnalysisConfig::EnableTensorRtEngine(int workspace_size, ...@@ -127,6 +127,7 @@ void contrib::AnalysisConfig::EnableTensorRtEngine(int workspace_size,
use_tensorrt_ = true; use_tensorrt_ = true;
tensorrt_workspace_size_ = workspace_size; tensorrt_workspace_size_ = workspace_size;
tensorrt_max_batchsize_ = max_batch_size; tensorrt_max_batchsize_ = max_batch_size;
tensorrt_min_subgraph_size_ = min_subgraph_size;
Update(); Update();
} }
...@@ -145,8 +146,8 @@ void contrib::AnalysisConfig::Update() { ...@@ -145,8 +146,8 @@ void contrib::AnalysisConfig::Update() {
LOG(ERROR) LOG(ERROR)
<< "TensorRT engine is not available when EnableGpu() not actived."; << "TensorRT engine is not available when EnableGpu() not actived.";
} else { } else {
// Append after the infer_clean pass. // Append after the Affine_channel_conv_fuse pass.
pass_builder()->InsertPass(1, "tensorrt_subgraph_pass"); pass_builder()->InsertPass(3, "tensorrt_subgraph_pass");
} }
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册