Merge pull request #12805 from tensor-tang/fix/op/elewise_add

luotao1 · web-flow · commit 8f9f414a1420 · 2018-08-20T21:41:17.000+08:00
fix SEGV element wise add at debug mode
diff --git a/paddle/fluid/operators/elementwise_op_function.h b/paddle/fluid/operators/elementwise_op_function.h
@@ -80,6 +80,9 @@ inline framework::DDim trim_trailing_singular_dims(
   for (int i = 0; i < actual_dims_size; ++i) {
     trim_dims[i] = dims[i];
   }
+  if (trim_dims.size() == 0) {
+    return framework::DDim(framework::make_dim());
+  }
   framework::DDim actual_dims = framework::make_ddim(trim_dims);
   return actual_dims;
 }

Original file line number	Diff line number	Diff line change
`@@ -80,6 +80,9 @@ inline framework::DDim trim_trailing_singular_dims(`
`80`	`80`	`for (int i = 0; i < actual_dims_size; ++i) {`
`81`	`81`	`trim_dims[i] = dims[i];`
`82`	`82`	`}`
	`83`	`+ if (trim_dims.size() == 0) {`
	`84`	`+ return framework::DDim(framework::make_dim());`
	`85`	`+ }`
`83`	`86`	`framework::DDim actual_dims = framework::make_ddim(trim_dims);`
`84`	`87`	`return actual_dims;`
`85`	`88`	`}`