From 8ea325aa5393a4ae43d8260f5532cca2cd63d78f Mon Sep 17 00:00:00 2001 From: HappyAngel Date: Sun, 26 Apr 2020 22:21:40 -0500 Subject: [PATCH] c++ API ( average_accumulates, tensor_array_to_tensor and average_accumulates) error message enhance. test=develop (#23631) --- .../fluid/operators/average_accumulates_op.cc | 69 ++++++++----------- .../get_tensor_from_selected_rows_op.cc | 34 +++++---- .../operators/tensor_array_to_tensor_op.cc | 10 ++- 3 files changed, 54 insertions(+), 59 deletions(-) diff --git a/paddle/fluid/operators/average_accumulates_op.cc b/paddle/fluid/operators/average_accumulates_op.cc index e2cde218db3..47c54c98fe2 100644 --- a/paddle/fluid/operators/average_accumulates_op.cc +++ b/paddle/fluid/operators/average_accumulates_op.cc @@ -48,48 +48,33 @@ class AverageAccumulatesOp : public framework::OperatorWithKernel { using framework::OperatorWithKernel::OperatorWithKernel; void InferShape(framework::InferShapeContext* ctx) const override { - PADDLE_ENFORCE( - ctx->HasInput("param"), - "Input (param) of average_accumulates op should not be null."); - PADDLE_ENFORCE( - ctx->HasInput("in_sum_1"), - "Input (sum_1) of average_accumulates op should not be null."); - PADDLE_ENFORCE( - ctx->HasInput("in_sum_2"), - "Input (sum_2) of average_accumulates op should not be null."); - PADDLE_ENFORCE( - ctx->HasInput("in_sum_3"), - "Input (sum_3) of average_accumulates op should not be null."); - PADDLE_ENFORCE( - ctx->HasInput("in_num_accumulates"), - "Input (in_num_accumulates) of average_accumulates op should " - "not be null."); - PADDLE_ENFORCE(ctx->HasInput("in_old_num_accumulates"), - "Input (old_num_accumulates) of average_accumulates op " - "should not be null."); - PADDLE_ENFORCE( - ctx->HasInput("in_num_updates"), - "Input (num_updates) of average_accumulates op should not be null."); - - PADDLE_ENFORCE( - ctx->HasOutput("out_sum_1"), - "Output (sum_1) of average_accumulates op should not be null."); - PADDLE_ENFORCE( - ctx->HasOutput("out_sum_2"), - "Output (sum_2) of average_accumulates op should not be null."); - PADDLE_ENFORCE( - ctx->HasOutput("out_sum_3"), - "Output (sum_3) of average_accumulates op should not be null."); - PADDLE_ENFORCE(ctx->HasOutput("out_num_accumulates"), - "Output (num_accumulates) of average_accumulates op should " - "not be null."); - PADDLE_ENFORCE(ctx->HasOutput("out_old_num_accumulates"), - "Output (old_num_accumulates) of average_accumulates op " - "should not be null."); - PADDLE_ENFORCE( - ctx->HasOutput("out_num_updates"), - "Output (num_updates) of average_accumulates op should not be null."); - + OP_INOUT_CHECK(ctx->HasInput("param"), "Input", "param", + "AverageAccumulates"); + OP_INOUT_CHECK(ctx->HasInput("in_sum_1"), "Input", "in_sum_1", + "AverageAccumulates"); + OP_INOUT_CHECK(ctx->HasInput("in_sum_2"), "Input", "in_sum_2", + "AverageAccumulates"); + OP_INOUT_CHECK(ctx->HasInput("in_sum_3"), "Input", "in_sum_3", + "AverageAccumulates"); + OP_INOUT_CHECK(ctx->HasInput("in_num_accumulates"), "Input", + "in_num_accumulates", "AverageAccumulates"); + OP_INOUT_CHECK(ctx->HasInput("in_old_num_accumulates"), "Input", + "in_old_num_accumulates", "AverageAccumulates"); + OP_INOUT_CHECK(ctx->HasInput("in_num_updates"), "Input", "in_num_updates", + "AverageAccumulates"); + + OP_INOUT_CHECK(ctx->HasOutput("out_sum_1"), "Output", "out_sum_1", + "AverageAccumulates"); + OP_INOUT_CHECK(ctx->HasOutput("out_sum_2"), "Output", "out_sum_2", + "AverageAccumulates"); + OP_INOUT_CHECK(ctx->HasOutput("out_sum_3"), "Output", "out_sum_3", + "AverageAccumulates"); + OP_INOUT_CHECK(ctx->HasOutput("out_num_accumulates"), "Output", + "out_num_accumulates", "AverageAccumulates"); + OP_INOUT_CHECK(ctx->HasOutput("out_old_num_accumulates"), "Output", + "out_old_num_accumulates", "AverageAccumulates"); + OP_INOUT_CHECK(ctx->HasOutput("out_num_updates"), "Output", + "out_num_updates", "AverageAccumulates"); auto in_dim = ctx->GetInputDim("param"); ctx->SetOutputDim("out_sum_1", in_dim); diff --git a/paddle/fluid/operators/get_tensor_from_selected_rows_op.cc b/paddle/fluid/operators/get_tensor_from_selected_rows_op.cc index 8b4dec13cfb..89a5d81a227 100644 --- a/paddle/fluid/operators/get_tensor_from_selected_rows_op.cc +++ b/paddle/fluid/operators/get_tensor_from_selected_rows_op.cc @@ -23,21 +23,25 @@ class GetTensorFromSelectedRowsOp : public framework::OperatorWithKernel { using framework::OperatorWithKernel::OperatorWithKernel; void InferShape(framework::InferShapeContext *ctx) const override { - PADDLE_ENFORCE(ctx->HasInput("X"), - "GetTensorFromSelectedRowsOp must have input X."); - PADDLE_ENFORCE(ctx->HasOutput("Out"), - "GetTensorFromSelectedRowsOp must have output Out."); - PADDLE_ENFORCE( - ctx->GetInputsVarType("X").front() == - framework::proto::VarType::SELECTED_ROWS, - "The input X's type should be SelectedRows, but the received is %s", - ctx->Inputs("X").front(), ctx->GetInputsVarType("X").front()); - PADDLE_ENFORCE( - ctx->GetOutputsVarType("Out").front() == - framework::proto::VarType::LOD_TENSOR, - "The output Out's type should be LoDTensor, but the received is %s", - ctx->Outputs("Out").front(), ctx->GetOutputsVarType("Out").front()); - + OP_INOUT_CHECK(ctx->HasInput("X"), "Input", "X", + "GetTensorFromSelectedRows"); + OP_INOUT_CHECK(ctx->HasOutput("Out"), "Output", "Out", + "GetTensorFromSelectedRows"); + + PADDLE_ENFORCE_EQ( + ctx->GetInputsVarType("X").front(), + framework::proto::VarType::SELECTED_ROWS, + platform::errors::InvalidArgument( + "The input X(%s)'s type should be SelectedRows, " + "but the received is %s", + ctx->Inputs("X").front(), ctx->GetInputsVarType("X").front())); + PADDLE_ENFORCE_EQ(ctx->GetOutputsVarType("Out").front(), + framework::proto::VarType::LOD_TENSOR, + platform::errors::InvalidArgument( + "The output Out(%s)'s type should be LoDTensor, " + "but the received is %s", + ctx->Outputs("Out").front(), + ctx->GetOutputsVarType("Out").front())); ctx->SetOutputDim("Out", ctx->GetInputDim("X")); } diff --git a/paddle/fluid/operators/tensor_array_to_tensor_op.cc b/paddle/fluid/operators/tensor_array_to_tensor_op.cc index 8a1621f58ab..620231eb2e2 100644 --- a/paddle/fluid/operators/tensor_array_to_tensor_op.cc +++ b/paddle/fluid/operators/tensor_array_to_tensor_op.cc @@ -96,7 +96,10 @@ class LoDTensorArray2TensorOp : public framework::OperatorBase { *scope.FindVar(Output("OutIndex"))->GetMutable(); const size_t n = inx.size(); - PADDLE_ENFORCE_GT(n, 0, "Input tensorarray size should > 0."); + PADDLE_ENFORCE_GT(n, 0, platform::errors::InvalidArgument( + "Input tensorarray size should > 0," + "but the received is %d", + n)); std::string base_name = Inputs("X")[0]; std::vector names; @@ -232,7 +235,10 @@ class LoDTensorArray2TensorGradOp : public framework::OperatorBase { auto &inx = scope.FindVar(Input("X"))->Get(); const size_t n = inx.size(); - PADDLE_ENFORCE_GT(n, 0, "Input tensorarray size should > 0."); + PADDLE_ENFORCE_GT(n, 0, platform::errors::InvalidArgument( + "Input tensorarray size should > 0, " + "but the received is: %d. ", + n)); std::string base_name = Inputs("X")[0]; std::vector names; -- GitLab