diff --git a/paddle/fluid/operators/fc_op.cc b/paddle/fluid/operators/fc_op.cc index 5bee0e480ed9304e6ab9179fa69ada374fe2bbee..72287ae6ac60f8de5eb62733791b0c9353dbe86b 100644 --- a/paddle/fluid/operators/fc_op.cc +++ b/paddle/fluid/operators/fc_op.cc @@ -134,6 +134,8 @@ class FCOpKernel : public framework::OpKernel { math::FCCompute( blas, in_dims[0], w_dims[1], w_dims[0], input_data, w_data, output_data, bias ? bias->data() : NULL); + + // TODO(TJ): fuse act } }; diff --git a/paddle/fluid/operators/fusion_lstm_op.cc b/paddle/fluid/operators/fusion_lstm_op.cc index b937f2edf19e0c9df4d2cf565cd286346546e0c3..3888333ec5626f1d8d35db215085f483c985cf0a 100644 --- a/paddle/fluid/operators/fusion_lstm_op.cc +++ b/paddle/fluid/operators/fusion_lstm_op.cc @@ -242,6 +242,7 @@ class FuisonLSTMKernel : public framework::OpKernel { to_batch(dev_ctx, *xx, batched_gate, true, is_reverse); } else { to_batch(dev_ctx, *x, xx, true, is_reverse); + batched_gate->set_lod(xx->lod()); math::FCCompute(blas, x_dims[0], wx_dims[1], x_dims[1], xx_data, wx_data, batched_gate_data, bias->data()); diff --git a/python/paddle/fluid/tests/unittests/test_fusion_lstm_op.py b/python/paddle/fluid/tests/unittests/test_fusion_lstm_op.py index 163e5cf6f80c77d9e17dbc3e2a7dea8fa6c69cf7..381edbb3c12cc92bbc1fe4dbe6a5ea1b73a76c4e 100644 --- a/python/paddle/fluid/tests/unittests/test_fusion_lstm_op.py +++ b/python/paddle/fluid/tests/unittests/test_fusion_lstm_op.py @@ -112,7 +112,7 @@ class TestLstmOp(OpTest): } def test_check_output(self): - self.check_output(atol=1e-8) + self.check_output() class TestLstmOpInitReverse(TestLstmOp):