提交 26b228e4 编写于 作者: T tangwei12

remove assignment and add vlog

上级 125e9166
...@@ -34,15 +34,15 @@ ...@@ -34,15 +34,15 @@
namespace paddle { namespace paddle {
namespace operators { namespace operators {
using paddle::framework::Tensor;
using paddle::platform::MKLDNNDeviceContext;
using paddle::platform::CPUDeviceContext;
using framework::DataLayout; using framework::DataLayout;
using mkldnn::memory; using mkldnn::memory;
using mkldnn::primitive; using mkldnn::primitive;
using mkldnn::reorder;
using mkldnn::stream; using mkldnn::stream;
using mkldnn::sum; using mkldnn::sum;
using mkldnn::reorder; using paddle::framework::Tensor;
using paddle::platform::CPUDeviceContext;
using paddle::platform::MKLDNNDeviceContext;
using platform::to_void_cast; using platform::to_void_cast;
template <typename T> template <typename T>
...@@ -186,8 +186,9 @@ class SumMKLDNNOpKernel : public paddle::framework::OpKernel<T> { ...@@ -186,8 +186,9 @@ class SumMKLDNNOpKernel : public paddle::framework::OpKernel<T> {
} }
if (in_dim.empty()) { if (in_dim.empty()) {
in_dim = framework::vectorize(get_selected_row(N - 1).value().dims()); VLOG(3) << "WARNING: all the inputs are empty"
} }
in_dim[0] = static_cast<int64_t>(first_dim); in_dim[0] = static_cast<int64_t>(first_dim);
out_value->Resize(framework::make_ddim(in_dim)); out_value->Resize(framework::make_ddim(in_dim));
......
...@@ -115,7 +115,7 @@ class SumKernel : public framework::OpKernel<T> { ...@@ -115,7 +115,7 @@ class SumKernel : public framework::OpKernel<T> {
} }
} }
if (in_dim.empty()) { if (in_dim.empty()) {
in_dim = framework::vectorize(get_selected_row(N - 1).value().dims()); VLOG(3) << "WARNING: all the inputs are empty"
} }
in_dim[0] = static_cast<int64_t>(first_dim); in_dim[0] = static_cast<int64_t>(first_dim);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册