Skip to content

Commit 3e876b3

Browse files
author
Tomasz Patejko
committed
MKL optimized elementwise add: fix style check
1 parent 9241011 commit 3e876b3

File tree

1 file changed

+11
-13
lines changed

1 file changed

+11
-13
lines changed

paddle/fluid/operators/elementwise_add_op.h

Lines changed: 11 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ struct IdentityGrad {
8585
HOSTDEVICE T operator()(T x, T y, T out, T dout) const { return dout; }
8686
};
8787

88-
template<typename DeviceContext, typename T>
88+
template <typename DeviceContext, typename T>
8989
void default_elementwise_add_grad(const framework::ExecutionContext& ctx,
9090
const framework::Tensor* x,
9191
const framework::Tensor* y,
@@ -100,16 +100,15 @@ void default_elementwise_add_grad(const framework::ExecutionContext& ctx,
100100
IdentityGrad<T>());
101101
}
102102

103-
template<typename DeviceContext, typename T>
103+
template <typename DeviceContext, typename T>
104104
typename std::enable_if<
105105
std::is_floating_point<T>::value &&
106106
std::is_same<DeviceContext, platform::CPUDeviceContext>::value>::type
107107
elementwise_add_grad(const framework::ExecutionContext& ctx,
108-
const framework::Tensor* x,
109-
const framework::Tensor* y,
108+
const framework::Tensor* x, const framework::Tensor* y,
110109
const framework::Tensor* out,
111-
const framework::Tensor* dout,
112-
framework::Tensor* dx, framework::Tensor* dy) {
110+
const framework::Tensor* dout, framework::Tensor* dx,
111+
framework::Tensor* dy) {
113112
auto blas = math::GetBlas<DeviceContext, T>(ctx);
114113

115114
if (dx) {
@@ -123,16 +122,15 @@ elementwise_add_grad(const framework::ExecutionContext& ctx,
123122
}
124123
}
125124

126-
template<typename DeviceContext, typename T>
125+
template <typename DeviceContext, typename T>
127126
typename std::enable_if<
128127
!std::is_floating_point<T>::value ||
129128
!std::is_same<DeviceContext, platform::CPUDeviceContext>::value>::type
130129
elementwise_add_grad(const framework::ExecutionContext& ctx,
131-
const framework::Tensor* x,
132-
const framework::Tensor* y,
130+
const framework::Tensor* x, const framework::Tensor* y,
133131
const framework::Tensor* out,
134-
const framework::Tensor* dout,
135-
framework::Tensor* dx, framework::Tensor* dy) {
132+
const framework::Tensor* dout, framework::Tensor* dx,
133+
framework::Tensor* dy) {
136134
default_elementwise_add_grad<DeviceContext, T>(ctx, x, y, out, dout, dx, dy);
137135
}
138136

@@ -152,8 +150,8 @@ class ElementwiseAddGradKernel : public framework::OpKernel<T> {
152150
if (platform::is_cpu_place(ctx.GetPlace()) && (x->dims() == y->dims())) {
153151
elementwise_add_grad<DeviceContext, T>(ctx, x, y, out, dout, dx, dy);
154152
} else {
155-
default_elementwise_add_grad<DeviceContext, T>(
156-
ctx, x, y, out, dout, dx, dy);
153+
default_elementwise_add_grad<DeviceContext, T>(ctx, x, y, out, dout, dx,
154+
dy);
157155
}
158156
}
159157
};

0 commit comments

Comments
 (0)