From 6828f3684bc3db37ec02b80c20b1e3537823baf3 Mon Sep 17 00:00:00 2001 From: Zeng Jinle <32832641+sneaxiy@users.noreply.github.com> Date: Wed, 11 Dec 2019 02:17:01 -0600 Subject: [PATCH] fix op_registry, add ignore op_function_impl.h, test=develop (#21654) --- paddle/fluid/framework/details/op_registry.h | 4 ++-- paddle/fluid/pybind/.gitignore | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/framework/details/op_registry.h b/paddle/fluid/framework/details/op_registry.h index ece0dc4e77b..3eb88ce84f8 100644 --- a/paddle/fluid/framework/details/op_registry.h +++ b/paddle/fluid/framework/details/op_registry.h @@ -170,8 +170,8 @@ struct OpInfoFiller { platform::errors::AlreadyExists( "Duplicate InferShapeFN of %s has been registered", op_type)); - auto* op = - dynamic_cast(info->creator_("", {}, {}, {})); + OperatorWithKernel* op = dynamic_cast(info->creator_( + std::string{}, VariableNameMap{}, VariableNameMap{}, AttributeMap{})); PADDLE_ENFORCE_NOT_NULL(op, platform::errors::InvalidArgument( "%s should have kernels", op_type)); info->infer_shape_ = [op](InferShapeContext* ctx) { diff --git a/paddle/fluid/pybind/.gitignore b/paddle/fluid/pybind/.gitignore index 8f222791edb..d86562fe73c 100644 --- a/paddle/fluid/pybind/.gitignore +++ b/paddle/fluid/pybind/.gitignore @@ -1 +1,2 @@ pybind.h +op_function_impl.h \ No newline at end of file -- GitLab