From 20f8741d289704f1557d6d8b70e07d704abb8ee8 Mon Sep 17 00:00:00 2001 From: seiriosPlus Date: Sun, 27 Sep 2020 16:49:11 +0800 Subject: [PATCH] fix error --- paddle/fluid/operators/distributed_ops/split_ids_op.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/operators/distributed_ops/split_ids_op.cc b/paddle/fluid/operators/distributed_ops/split_ids_op.cc index 0fe8f58595f..86738b7c69e 100644 --- a/paddle/fluid/operators/distributed_ops/split_ids_op.cc +++ b/paddle/fluid/operators/distributed_ops/split_ids_op.cc @@ -52,8 +52,8 @@ class SplitIdsOp : public framework::OperatorWithKernel { using framework::OperatorWithKernel::OperatorWithKernel; void InferShape(framework::InferShapeContext *ctx) const override { - OP_INOUT_CHECK(ctx->HasInput("Ids"), "Input", "Ids", "SplitIdsOp"); - OP_INOUT_CHECK(ctx->HasOutput("Out"), "Output", "Out", "SplitIdsOp"); + OP_INOUT_CHECK(ctx->HasInputs("Ids"), "Input", "Ids", "SplitIdsOp"); + OP_INOUT_CHECK(ctx->HasOutputs("Out"), "Output", "Out", "SplitIdsOp"); auto ids_var_type = ctx->GetInputsVarType("Ids").front(); auto ids_dims = ctx->GetInputsDim("Ids"); -- GitLab