From 0507f7bc3c9caf7c527446b710d22d6fdae7acc8 Mon Sep 17 00:00:00 2001 From: tensor-tang Date: Mon, 20 Aug 2018 14:34:59 +0800 Subject: [PATCH] fix SEGV elementwise add at debug mode --- paddle/fluid/operators/elementwise_op_function.h | 3 +++ 1 file changed, 3 insertions(+) diff --git a/paddle/fluid/operators/elementwise_op_function.h b/paddle/fluid/operators/elementwise_op_function.h index bc3e95e904f..f90dcdc1565 100644 --- 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; } -- GitLab