Fix for accuracy problem for inplace operators when MKL-DNN mode is enabled

test=develop
6 years ago · bc7db6cec9
parent dfd4a111cb
commit bc7db6cec9
1 changed files with 2 additions and 1 deletions
--- a/paddle/fluid/framework/operator.cc
+++ b/paddle/fluid/framework/operator.cc
@ -749,7 +749,7 @@ void OperatorWithKernel::RunImpl(const Scope& scope,

  kernel_iter->second(ExecutionContext(*this, exec_scope, *dev_ctx));

-  if (!transfered_inplace_vars.empty()) {
+  if (run_by_executor_ && !transfered_inplace_vars.empty()) {
    // there is inplace variable has been transfered.
    TransferInplaceVarsBack(scope, transfered_inplace_vars, *transfer_scope);
  }
@ -771,6 +771,7 @@ void OperatorWithKernel::RunImpl(const Scope& scope,
    }
  }
 }
+
 void OperatorWithKernel::TransferInplaceVarsBack(
    const Scope& scope, const std::vector<std::string>& inplace_vars,
    const Scope& transfer_scope) const {