Register op version for linspace,test=op_version (#30025)

* Register op version for linspace,test=op_version * Register op version for linspace,test=op_version * Register op version for linspace,test=op_version * Register op version for linspace,test=op_version * Register op version for linspace,test=op_version
4 years ago · 6e93fb92f9
parent 9e51e3833f
commit 6e93fb92f9
1 changed files with 9 additions and 0 deletions
--- a/paddle/fluid/operators/linspace_op.cc
+++ b/paddle/fluid/operators/linspace_op.cc
@ -14,6 +14,7 @@ limitations under the License. */

 #include "paddle/fluid/operators/linspace_op.h"
 #include <string>
+#include "paddle/fluid/framework/op_version_registry.h"

 namespace paddle {
 namespace operators {
@ -92,3 +93,11 @@ REGISTER_OP_CPU_KERNEL(linspace, ops::CPULinspaceKernel<float>,
                       ops::CPULinspaceKernel<int32_t>,
                       ops::CPULinspaceKernel<int64_t>,
                       ops::CPULinspaceKernel<double>);
+
+REGISTER_OP_VERSION(linspace)
+    .AddCheckpoint(
+        R"ROC(
+      Upgrade linspace to add a new attribute [dtype].
+    )ROC",
+        paddle::framework::compatible::OpVersionDesc().NewAttr(
+            "dtype", "In order to change output data type ", 5));