From 99e333736865df8b6b4faa0b8ae94a1fec6a09f9 Mon Sep 17 00:00:00 2001 From: xiemoyuan <71377852+xiemoyuan@users.noreply.github.com> Date: Mon, 28 Sep 2020 14:52:10 +0800 Subject: [PATCH] Optimize the error message of OP. (#27478) * iCafe 9009: Optimize the error message of OP. * Optimize the error message of GatherTreeOP. --- paddle/fluid/operators/gather_tree_op.cc | 16 +++++++--------- 1 file changed, 7 insertions(+), 9 deletions(-) diff --git a/paddle/fluid/operators/gather_tree_op.cc b/paddle/fluid/operators/gather_tree_op.cc index 26f9989121e..830134e57e0 100644 --- a/paddle/fluid/operators/gather_tree_op.cc +++ b/paddle/fluid/operators/gather_tree_op.cc @@ -22,18 +22,16 @@ class GatherTreeOp : public framework::OperatorWithKernel { using framework::OperatorWithKernel::OperatorWithKernel; void InferShape(framework::InferShapeContext* ctx) const override { - PADDLE_ENFORCE(ctx->HasInput("Ids"), - "Input(Ids) of GatherTreeOp should not be null."); - PADDLE_ENFORCE(ctx->HasInput("Parents"), - "Input(Parents) of GatherTreeOp should not be null."); - PADDLE_ENFORCE(ctx->HasOutput("Out"), - "Output(Out) of GatherTreeOp should not be null."); + OP_INOUT_CHECK(ctx->HasInput("Ids"), "Input", "Ids", "GatherTree"); + OP_INOUT_CHECK(ctx->HasInput("Parents"), "Input", "Parents", "GatherTree"); + OP_INOUT_CHECK(ctx->HasOutput("Out"), "Output", "Out", "GatherTree"); auto ids_dims = ctx->GetInputDim("Ids"); auto parents_dims = ctx->GetInputDim("Parents"); - PADDLE_ENFORCE(ids_dims == parents_dims, - "The shape of Input(Parents) must be same with the shape of " - "Input(Ids)."); + PADDLE_ENFORCE_EQ(ids_dims == parents_dims, true, + platform::errors::InvalidArgument( + "The shape of Input(Parents) must be same with the " + "shape of Input(Ids).")); ctx->SetOutputDim("Out", ids_dims); } -- GitLab