未验证 提交 96e58f87 编写于 作者: W Wilber 提交者: GitHub

fix for mkldnn (#48852)

上级 6e356c68
......@@ -948,12 +948,7 @@ void AnalysisConfig::Update() {
#endif
}
#ifdef PADDLE_WITH_MKLDNN
// Do not optimize when mkldnn is on
if (enable_memory_optim_ && !use_mkldnn_) {
#else
if (enable_memory_optim_) {
#endif
pass_builder()->AppendAnalysisPass("memory_optimize_pass");
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册