Skip to content

Commit f7104b8

Browse files
committed
Fixed issues with phi kernel
1 parent 1cdf6a3 commit f7104b8

File tree

4 files changed

+10
-10
lines changed

4 files changed

+10
-10
lines changed

paddle/phi/kernels/activation_grad_kernel.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -125,18 +125,18 @@ void EluDoubleGradKernel(const Context& dev_ctx,
125125
template <typename T, typename Context>
126126
void SigmoidDoubleGradKernel(const Context& dev_ctx,
127127
const DenseTensor& out,
128-
const DenseTensor& ddx,
129128
const DenseTensor& dout,
129+
const DenseTensor& ddx,
130130
DenseTensor* dout_new,
131131
DenseTensor* ddout);
132132

133133
template <typename T, typename Context>
134134
void SigmoidTripleGradKernel(const Context& dev_ctx,
135135
const DenseTensor& out,
136-
const DenseTensor& ddx,
137136
const DenseTensor& dout,
138-
const DenseTensor& d_ddout,
137+
const DenseTensor& ddx,
139138
const DenseTensor& d_dout_new,
139+
const DenseTensor& d_ddout,
140140
DenseTensor* d_out_new,
141141
DenseTensor* d_dout,
142142
DenseTensor* d_ddx);

paddle/phi/kernels/funcs/activation_functor.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1414,12 +1414,12 @@ struct SigmoidTripleGradFunctor : public BaseActivationFunctor<T> {
14141414
template <typename Device>
14151415
void operator()(const Device& dev,
14161416
const DenseTensor* Out,
1417-
const DenseTensor* dOut,
14181417
const DenseTensor* ddX,
1419-
const DenseTensor* d_dOut_New,
1418+
const DenseTensor* dOut,
14201419
const DenseTensor* d_DDOut,
1421-
DenseTensor* d_Out_New,
1420+
const DenseTensor* d_dOut_New,
14221421
DenseTensor* d_d_Out,
1422+
DenseTensor* d_Out_New,
14231423
DenseTensor* d_DDx) const {
14241424
auto* d = dev.eigen_device();
14251425
auto ddx = EigenVector<T>::Flatten(

paddle/phi/kernels/impl/activation_grad_impl.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -262,10 +262,10 @@ void SigmoidDoubleGradKernel(const Context& dev_ctx,
262262
template <typename T, typename Context>
263263
void SigmoidTripleGradKernel(const Context& dev_ctx,
264264
const DenseTensor& out,
265-
const DenseTensor& ddx,
266265
const DenseTensor& dout,
267-
const DenseTensor& d_ddout,
266+
const DenseTensor& ddx,
268267
const DenseTensor& d_dout_new,
268+
const DenseTensor& d_ddout,
269269
DenseTensor* d_out_new,
270270
DenseTensor* d_dout,
271271
DenseTensor* d_ddx) {

paddle/phi/ops/compat/activation_sig.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -139,13 +139,13 @@ KernelSignature TanhTripleGradOpArgumentMapping(
139139
KernelSignature SigmoidDoubleGradOpArgumentMapping(
140140
const ArgumentMappingContext& ctx) {
141141
return KernelSignature(
142-
"sigmoid_double_grad", {"Out", "DDX", "DOut"}, {}, {"DOutNew", "DDOut"});
142+
"sigmoid_double_grad", {"Out", "DOut", "DDX"}, {}, {"DOutNew", "DDOut"});
143143
}
144144

145145
KernelSignature SigmoidTripleGradOpArgumentMapping(
146146
const ArgumentMappingContext& ctx) {
147147
return KernelSignature("sigmoid_triple_grad",
148-
{"Out", "DDX", "DOut", "D_DDOut", "D_DOut_New"},
148+
{"Out", "DOut", "DDX", "D_DOut_New", "D_DDOut"},
149149
{},
150150
{"D_OutNew", "D_DOut", "D_DDx"});
151151
}

0 commit comments

Comments
 (0)