From 869f00ffc6697bdac73271ecbd7257f6937245c2 Mon Sep 17 00:00:00 2001 From: liuhongyu Date: Thu, 14 Feb 2019 16:20:37 +0800 Subject: [PATCH] set lstm lstmp unsed pointer to null --- paddle/fluid/operators/lstm_op.h | 4 ++++ paddle/fluid/operators/lstmp_op.h | 5 +++++ 2 files changed, 9 insertions(+) diff --git a/paddle/fluid/operators/lstm_op.h b/paddle/fluid/operators/lstm_op.h index 7d62d2d020..289f50f52e 100644 --- a/paddle/fluid/operators/lstm_op.h +++ b/paddle/fluid/operators/lstm_op.h @@ -311,6 +311,10 @@ class LSTMGradKernel : public framework::OpKernel { lstm_grad.prev_state_grad = c0_g ? ordered_c0_g.data() : nullptr; } + // lstm_value.output_value not used in bp, set to null + // lstm_grad.state_active_grad not used in bp, set to null + lstm_value.output_value = nullptr; + lstm_grad.state_active_grad = nullptr; int cur_batch_size = bend - bstart; math::LstmUnitGradFunctor::compute( device_ctx, lstm_value, lstm_grad, frame_size, cur_batch_size, diff --git a/paddle/fluid/operators/lstmp_op.h b/paddle/fluid/operators/lstmp_op.h index 370dd04d14..05ecd3c1ae 100644 --- a/paddle/fluid/operators/lstmp_op.h +++ b/paddle/fluid/operators/lstmp_op.h @@ -405,6 +405,11 @@ class LSTMPGradKernel : public framework::OpKernel { } int cur_batch_size = bend - bstart; + // lstm_value.output_value not used in bp, set to null + // lstm_grad.state_active_grad not used in bp, set to null + lstm_value.output_value = nullptr; + lstm_grad.state_active_grad = nullptr; + math::LstmUnitGradFunctor::compute( device_ctx, lstmp_value, lstmp_grad, frame_size, cur_batch_size, gate_act, cell_act, cand_act); -- GitLab