Fix sequence_softmax_cudnn op

eb8fd853 · Yu Yang · 37686779 · eb8fd853
隐藏空白更改
内联并排

Showing with 3 addition and 1 deletion

paddle/fluid/operators/sequence_softmax_cudnn_op.cu.cc paddle/fluid/operators/sequence_softmax_cudnn_op.cu.cc +3 -1

未找到文件。
--- a/paddle/fluid/operators/sequence_softmax_cudnn_op.cu.cc
+++ b/paddle/fluid/operators/sequence_softmax_cudnn_op.cu.cc
@@ -68,7 +68,9 @@ class SequenceSoftmaxGradCUDNNKernel : public framework::OpKernel<T> {
    auto* out_grad = ctx.Input<LoDTensor>(framework::GradVarName("Out"));
    auto* x = ctx.Input<LoDTensor>("X");
    auto* x_grad = ctx.Output<LoDTensor>(framework::GradVarName("X"));
-
+    if (x_grad) {
+      x_grad->set_lod(x->lod());
+    }
    auto lod = x->lod();
    const size_t level = lod.size() - 1;