From 803b7b6298bea773316b3030b2edbc2fa183b738 Mon Sep 17 00:00:00 2001 From: fengjiayi Date: Fri, 6 Oct 2017 16:01:14 -0700 Subject: [PATCH] Follow comments --- paddle/framework/op_info.h | 2 ++ paddle/framework/op_registry.cc | 4 ++-- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/paddle/framework/op_info.h b/paddle/framework/op_info.h index 231f212fa3..c504f69e30 100644 --- a/paddle/framework/op_info.h +++ b/paddle/framework/op_info.h @@ -54,6 +54,8 @@ struct OpInfo { "Operator GradOpMaker has not been registered."); return grad_op_maker_; } + + const OpAttrChecker* Checker() const { return checker_; } }; class OpInfoMap { diff --git a/paddle/framework/op_registry.cc b/paddle/framework/op_registry.cc index e9d2e55872..504afbd5db 100644 --- a/paddle/framework/op_registry.cc +++ b/paddle/framework/op_registry.cc @@ -23,8 +23,8 @@ std::unique_ptr OpRegistry::CreateOp( const std::string& type, const VariableNameMap& inputs, const VariableNameMap& outputs, AttributeMap attrs) { auto& info = OpInfoMap::Instance().Get(type); - if (info.checker_ != nullptr) { - info.checker_->Check(attrs); + if (info.Checker() != nullptr) { + info.Checker()->Check(attrs); } auto op = info.Creator()(type, inputs, outputs, attrs); return std::unique_ptr(op); -- GitLab