Skip to content

Commit e79841d

Browse files
author
xutianbing
committed
fix test_LayerGrad.
1 parent 1251501 commit e79841d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

paddle/function/ContextProjectionOp.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,7 @@ class ContextProjectionForwardFunc : public FunctionBase {
132132
auto out_mat = out_seq.matrix<Device>();
133133
const auto in_mat = val_seqs.matrix<Device>();
134134
const auto w_mat =
135-
(2UL == inputs.size())
135+
(2UL == inputs.size() && inputs[1].data())
136136
? inputs[1].matrix<Device>()
137137
: typename Tensor<real, Device>::Matrix(nullptr, 0, 0);
138138
const auto seq_vec = val_seqs.getSequenceId().vector<int, Device>();
@@ -263,7 +263,7 @@ class ContextProjectionBackwardFunc : public FunctionBase {
263263
!out_seq.data() ? typename Tensor<real, Device>::Matrix(nullptr, 0, 0)
264264
: out_seq.matrix<Device>();
265265
auto w_grad_mat =
266-
(2UL == outputs.size())
266+
(2UL == outputs.size() && outputs[1].data())
267267
? outputs[1].matrix<Device>()
268268
: typename Tensor<real, Device>::Matrix(nullptr, 0, 0);
269269

0 commit comments

Comments
 (0)