diff --git a/paddle/fluid/operators/controlflow/compare_op.cc b/paddle/fluid/operators/controlflow/compare_op.cc index 21c28f9818b51e5b52ca2d89684f95d672a2d913..3cad86d96c26a0e25fcbaeb02405315895744e50 100644 --- 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) \