未验证 提交 ddcff254 编写于 作者: C channings 提交者: GitHub

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

Co-authored-by: Nzhoushunjie <zhoushunjie@baidu.com>
上级 66e16b7e
...@@ -133,9 +133,9 @@ class CompareOp : public framework::OperatorWithKernel { ...@@ -133,9 +133,9 @@ class CompareOp : public framework::OperatorWithKernel {
REGISTER_OP_VERSION(op_type) \ REGISTER_OP_VERSION(op_type) \
.AddCheckpoint( \ .AddCheckpoint( \
R"ROC(Upgrade compare ops, add a new attribute [force_cpu])ROC", \ R"ROC(Upgrade compare ops, add a new attribute [force_cpu])ROC", \
paddle::framework::compatible::OpVersionDesc().NewAttr( \ paddle::framework::compatible::OpVersionDesc().ModifyAttr( \
"force_cpu", \ "force_cpu", \
"In order to force fill output variable to cpu memory.", \ "In order to force fill output variable to gpu memory.", \
false)); false));
#define REGISTER_COMPARE_OP(op_type, _equation) \ #define REGISTER_COMPARE_OP(op_type, _equation) \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册