Follow comments

PaddlePaddle · Oct 6, 2017 · 803b7b6 · 803b7b6
1 parent 097f533
commit 803b7b6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git 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
@@ -23,8 +23,8 @@ std::unique_ptr<OperatorBase> 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<OperatorBase>(op);