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

luotao1 · web-flow · commit d1135906494c · 2018-07-17T22:25:18.000+08:00
MKLDNN sum fix: remove in_place condition in loop with memory primitives
diff --git 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>();

Original file line number	Diff line number	Diff line change
`@@ -88,7 +88,7 @@ class SumMKLDNNOpKernel : public paddle::framework::OpKernel<T> {`
`88`	`88`	`input_format = memory::format::nc;`
`89`	`89`	`}`
`90`	`90`
`91`		`- for (int i = in_place ? 1 : 0; i < N; i++) {`
	`91`	`+ for (int i = 0; i < N; i++) {`
`92`	`92`	`PADDLE_ENFORCE(in_vars[i]->IsType<LoDTensor>(),`
`93`	`93`	`"all inputs must be all LoDTensors");`
`94`	`94`	`auto& input = in_vars[i]->Get<LoDTensor>();`