未验证 提交 d1135906 编写于 作者: T Tao Luo 提交者: GitHub

Merge pull request #12180 from tpatejko/tpatejko/edit-distance-fix-in-mkldnn-sum

MKLDNN sum fix: remove in_place condition in loop with memory primitives
......@@ -88,7 +88,7 @@ class SumMKLDNNOpKernel : public paddle::framework::OpKernel<T> {
input_format = memory::format::nc;
}
for (int i = in_place ? 1 : 0; i < N; i++) {
for (int i = 0; i < N; i++) {
PADDLE_ENFORCE(in_vars[i]->IsType<LoDTensor>(),
"all inputs must be all LoDTensors");
auto& input = in_vars[i]->Get<LoDTensor>();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册