update op_version_registry, test=develop

PaddlePaddle · Aug 24, 2020 · 03183d6 · 03183d6 · paddle-bot-old · Aug 24, 2020
1 parent 8645591
commit 03183d6
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 7 deletions.
diff --git a/paddle/fluid/framework/op_version_registry.h b/paddle/fluid/framework/op_version_registry.h
@@ -48,11 +48,15 @@ struct ModifyAttr : OpUpdateRecord {
   boost::any default_value_;
 };
 struct NewAttr : OpUpdateRecord {
-  NewAttr(const std::string& name, const std::string& remark)
-      : OpUpdateRecord({Type::kNewAttr, remark}), name_(name) {}
+  NewAttr(const std::string& name, const std::string& remark,
+          boost::any default_value)
+      : OpUpdateRecord({Type::kNewAttr, remark}),
+        name_(name),
+        default_value_(default_value) {}
 
  private:
   std::string name_;
+  boost::any default_value_;
 };
 
 class OpVersionDesc {
@@ -64,9 +68,10 @@ class OpVersionDesc {
     return *this;
   }
 
-  OpVersionDesc& NewAttr(const std::string& name, const std::string& remark) {
-    infos_.push_back(
-        std::shared_ptr<OpUpdateRecord>(new compatible::NewAttr(name, remark)));
+  OpVersionDesc& NewAttr(const std::string& name, const std::string& remark,
+                         boost::any default_value) {
+    infos_.push_back(std::shared_ptr<OpUpdateRecord>(
+        new compatible::NewAttr(name, remark, default_value)));
     return *this;
   }
 

diff --git a/paddle/fluid/framework/op_version_registry_test.cc b/paddle/fluid/framework/op_version_registry_test.cc
@@ -32,15 +32,17 @@ TEST(test_operator_version, test_operator_version) {
                           "Increased from the original one method to two.", -1)
               .NewAttr("size",
                        "In order to represent a two-dimensional rectangle, the "
-                       "parameter size is added."))
+                       "parameter size is added.",
+                       0))
       .AddCheckpoint(
           R"ROC(
         Add a new attribute [height]
       )ROC",
           framework::compatible::OpVersionDesc().NewAttr(
               "height",
               "In order to represent a two-dimensional rectangle, the "
-              "parameter height is added."));
+              "parameter height is added.",
+              0));
 }
 }  // namespace compatible
 }  // namespace framework