提交 5ecdc49c 编写于 作者: L luotao1

set enable_runtime_context_cache_ default false

test=develop
上级 46ee6bb1
...@@ -205,8 +205,6 @@ void AnalysisConfig::Update() { ...@@ -205,8 +205,6 @@ void AnalysisConfig::Update() {
// Append after the Affine_channel_conv_fuse pass. // Append after the Affine_channel_conv_fuse pass.
pass_builder()->InsertPass(3, "tensorrt_subgraph_pass"); pass_builder()->InsertPass(3, "tensorrt_subgraph_pass");
} }
// runtime_context_cache isn't fit for tensorrt.
enable_runtime_context_cache_ = false;
} }
if (use_mkldnn_) { if (use_mkldnn_) {
......
...@@ -278,7 +278,7 @@ struct AnalysisConfig { ...@@ -278,7 +278,7 @@ struct AnalysisConfig {
// since the input/output names of this Op do not change in the execution, // since the input/output names of this Op do not change in the execution,
// RuntimeContext could be created only at the first iteration of this Op's // RuntimeContext could be created only at the first iteration of this Op's
// execution to save the elapsed time. // execution to save the elapsed time.
bool enable_runtime_context_cache_{true}; bool enable_runtime_context_cache_{false};
// A runtime cache, shouldn't be transferred to others. // A runtime cache, shouldn't be transferred to others.
std::string serialized_info_cache_; std::string serialized_info_cache_;
......
...@@ -107,6 +107,7 @@ void SetConfig(AnalysisConfig *cfg) { ...@@ -107,6 +107,7 @@ void SetConfig(AnalysisConfig *cfg) {
cfg->DisableGpu(); cfg->DisableGpu();
cfg->SwitchSpecifyInputNames(); cfg->SwitchSpecifyInputNames();
cfg->SwitchIrOptim(); cfg->SwitchIrOptim();
cfg->SwitchRuntimeContextCache();
if (FLAGS_zero_copy) { if (FLAGS_zero_copy) {
cfg->SwitchUseFeedFetchOps(false); cfg->SwitchUseFeedFetchOps(false);
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册