fix op_register_version for compare ops, test=op_version (#30007)

Co-authored-by: zhoushunjie <zhoushunjie@baidu.com>
5 years ago · ddcff254db
parent 66e16b7e99
commit ddcff254db
1 changed files with 2 additions and 2 deletions
--- a/paddle/fluid/operators/controlflow/compare_op.cc
+++ b/paddle/fluid/operators/controlflow/compare_op.cc
@ -133,9 +133,9 @@ class CompareOp : public framework::OperatorWithKernel {
  REGISTER_OP_VERSION(op_type)                                             \
      .AddCheckpoint(                                                      \
          R"ROC(Upgrade compare ops, add a new attribute [force_cpu])ROC", \
-          paddle::framework::compatible::OpVersionDesc().NewAttr(          \
+          paddle::framework::compatible::OpVersionDesc().ModifyAttr(       \
              "force_cpu",                                                 \
-              "In order to force fill output variable to cpu memory.",     \
+              "In order to force fill output variable to gpu memory.",     \
              false));

 #define REGISTER_COMPARE_OP(op_type, _equation)                           \