MKLDNN sum fix: remove in_place condition in loop creating memory primitives for sum

7 years ago · b2b8b15bfe
parent e90678c051
commit b2b8b15bfe
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/operators/sum_mkldnn_op.cc
+++ b/paddle/fluid/operators/sum_mkldnn_op.cc
@ -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>();