From d42f93e504f2b07d26d5dea5d026f385a0aca33a Mon Sep 17 00:00:00 2001 From: FlyingQianMM <245467267@qq.com> Date: Tue, 29 Dec 2020 22:06:33 +0800 Subject: [PATCH] add op_register_version for allclose op; test=op_version (#29968) --- paddle/fluid/operators/allclose_op.cc | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/paddle/fluid/operators/allclose_op.cc b/paddle/fluid/operators/allclose_op.cc index cd83443f052..fe6c3c9adcc 100644 --- a/paddle/fluid/operators/allclose_op.cc +++ b/paddle/fluid/operators/allclose_op.cc @@ -15,6 +15,7 @@ #include "paddle/fluid/operators/allclose_op.h" #include #include "paddle/fluid/framework/op_registry.h" +#include "paddle/fluid/framework/op_version_registry.h" #include "paddle/fluid/framework/operator.h" #include "paddle/fluid/platform/enforce.h" @@ -153,3 +154,28 @@ REGISTER_OPERATOR( ops::AllcloseOpVarTypeInference); REGISTER_OP_CPU_KERNEL(allclose, ops::AllcloseKernel, ops::AllcloseKernel); + +/* ========================== register checkpoint ===========================*/ +REGISTER_OP_VERSION(allclose) + .AddCheckpoint( + R"ROC(Upgrade allclose, add two new inputs [Rtol] and [Atol].)ROC", + paddle::framework::compatible::OpVersionDesc() + .NewInput("Rtol", + "The added input 'Rtol' is not" + "dispensable.") + .NewInput("Atol", + "The added input 'Atol' is not" + "dispensable.")) + .AddCheckpoint( + R"ROC(Delete two attributes [rtol] and [atol])ROC", + paddle::framework::compatible::OpVersionDesc() + .DeleteAttr("rtol", + "The attribute 'rtol' is deleted." + "The reason why it is deleted is that" + "attributes do not support a float64 value" + "and it is changed to a tensor.") + .DeleteAttr("atol", + "The attribute 'atol' is deleted." + "The reason why it is deleted is that" + "attributes do not support a float64 value" + "and it is changed to a tensor.")); -- GitLab