Fix sum op fails as no memory in tensor(PaddlePaddle#20602)

test=develop Signed-off-by: zhaoyuchen <zhaoyuchen01@baidu.com>
seiriosPlus · Oct 15, 2019 · 0c22685 · 0c22685
1 parent 496ba7a
commit 0c22685
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/paddle/fluid/operators/sum_op.cu b/paddle/fluid/operators/sum_op.cu
@@ -163,8 +163,10 @@ void SumToLoDTensor(const framework::ExecutionContext &context) {
   for (int i = start; i < in_num; ++i) {
     if (in_vars[i]->IsType<framework::LoDTensor>()) {
       auto &in_i = in_vars[i]->Get<framework::LoDTensor>();
-      in_data.emplace_back(in_i.data<T>());
       lod_length = in_i.numel();
+      if (lod_length && in_i.IsInitialized()) {
+        in_data.emplace_back(in_i.data<T>());
+      }
     } else if (in_vars[i]->IsType<framework::SelectedRows>()) {
       selectrow_index.push_back(i);
     }