未验证 提交 8314e64a 编写于 作者: Z zhaoyuchen2018 提交者: GitHub

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

test=develop
Signed-off-by: Nzhaoyuchen <zhaoyuchen01@baidu.com>
上级 48b27229
...@@ -163,8 +163,10 @@ void SumToLoDTensor(const framework::ExecutionContext &context) { ...@@ -163,8 +163,10 @@ void SumToLoDTensor(const framework::ExecutionContext &context) {
for (int i = start; i < in_num; ++i) { for (int i = start; i < in_num; ++i) {
if (in_vars[i]->IsType<framework::LoDTensor>()) { if (in_vars[i]->IsType<framework::LoDTensor>()) {
auto &in_i = in_vars[i]->Get<framework::LoDTensor>(); auto &in_i = in_vars[i]->Get<framework::LoDTensor>();
in_data.emplace_back(in_i.data<T>());
lod_length = in_i.numel(); 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>()) { } else if (in_vars[i]->IsType<framework::SelectedRows>()) {
selectrow_index.push_back(i); selectrow_index.push_back(i);
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册