提交 996d12f1 编写于 作者: T Tomasz Patejko

MKL optimized elementwise add backward: coding style fixes

上级 5a622c29
...@@ -26,5 +26,5 @@ REGISTER_OP_CPU_KERNEL( ...@@ -26,5 +26,5 @@ REGISTER_OP_CPU_KERNEL(
elementwise_add_grad, elementwise_add_grad,
ops::ElementwiseAddGradKernel<paddle::platform::CPUDeviceContext, float>, ops::ElementwiseAddGradKernel<paddle::platform::CPUDeviceContext, float>,
ops::ElementwiseAddGradKernel<paddle::platform::CPUDeviceContext, double>); ops::ElementwiseAddGradKernel<paddle::platform::CPUDeviceContext, double>);
// ops::ElementwiseAddGradKernel<paddle::platform::CPUDeviceContext, int>, // ops::ElementwiseAddGradKernel<paddle::platform::CPUDeviceContext, int>,
// ops::ElementwiseAddGradKernel<paddle::platform::CPUDeviceContext, int64_t>); // ops::ElementwiseAddGradKernel<paddle::platform::CPUDeviceContext, int64_t>);
...@@ -102,7 +102,7 @@ class ElementwiseAddGradKernel : public framework::OpKernel<T> { ...@@ -102,7 +102,7 @@ class ElementwiseAddGradKernel : public framework::OpKernel<T> {
if (platform::is_cpu_place(ctx.GetPlace()) && (x->dims() == y->dims())) { if (platform::is_cpu_place(ctx.GetPlace()) && (x->dims() == y->dims())) {
auto blas = math::GetBlas<DeviceContext, T>(ctx); auto blas = math::GetBlas<DeviceContext, T>(ctx);
if (dx) if (dx)
dx->mutable_data<T>(ctx.GetPlace()); dx->mutable_data<T>(ctx.GetPlace());
if (dy) if (dy)
dy->mutable_data<T>(ctx.GetPlace()); dy->mutable_data<T>(ctx.GetPlace());
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册