提交 3ace486e 编写于 作者: Q Qiao Longfei

fix sum_op selected rows test=develop

上级 b16e832d
...@@ -52,10 +52,12 @@ class SumOp : public framework::OperatorWithKernel { ...@@ -52,10 +52,12 @@ class SumOp : public framework::OperatorWithKernel {
framework::DDim in_dim({0}); framework::DDim in_dim({0});
for (size_t i = 0; i < x_dims.size(); ++i) { for (size_t i = 0; i < x_dims.size(); ++i) {
if (x_var_types[i] == framework::proto::VarType::SELECTED_ROWS) { auto& x_dim = x_dims[i];
// x_dim.size() == 1 means the real dim of selected rows is [0]
if (x_var_types[i] == framework::proto::VarType::SELECTED_ROWS &&
x_dim.size() == 1) {
continue; continue;
} }
auto& x_dim = x_dims[i];
if (framework::product(x_dim) == 0) { if (framework::product(x_dim) == 0) {
continue; continue;
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册