提交 bc7db6ce 编写于 作者: K Krzysztof Binias

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

test=develop
上级 dfd4a111
......@@ -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 {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册