fix op_registry, add ignore op_function_impl.h, test=develop (#21654)

paddle_tiny_install
Zeng Jinle 6 years ago committed by GitHub
parent 45d2fa4e26
commit 6828f3684b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -170,8 +170,8 @@ struct OpInfoFiller<T, kOperator> {
platform::errors::AlreadyExists(
"Duplicate InferShapeFN of %s has been registered", op_type));
auto* op =
dynamic_cast<OperatorWithKernel*>(info->creator_("", {}, {}, {}));
OperatorWithKernel* op = dynamic_cast<OperatorWithKernel*>(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) {

@ -1 +1,2 @@
pybind.h
op_function_impl.h
Loading…
Cancel
Save