From e891f4da1b4e1689f2827f77261a60bd029f5682 Mon Sep 17 00:00:00 2001
From: yongqiangma <xing.wo@163.com>
Date: Mon, 4 Jan 2021 20:45:21 +0800
Subject: [PATCH] Add p_norm op version info (#30042)

* p_norm fix op version info. test=develop
---
 paddle/fluid/operators/p_norm_op.cc | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/paddle/fluid/operators/p_norm_op.cc b/paddle/fluid/operators/p_norm_op.cc
index 426a059c2a..7576950561 100644
--- a/paddle/fluid/operators/p_norm_op.cc
+++ b/paddle/fluid/operators/p_norm_op.cc
@@ -16,6 +16,7 @@ limitations under the License. */
 #include <memory>
 #include <string>
 #include <vector>
+#include "paddle/fluid/framework/op_version_registry.h"
 
 namespace paddle {
 namespace operators {
@@ -174,3 +175,11 @@ REGISTER_OP_CPU_KERNEL(p_norm, ops::PnormKernel<CPU, float>,
                        ops::PnormKernel<CPU, double>);
 REGISTER_OP_CPU_KERNEL(p_norm_grad, ops::PnormGradKernel<CPU, float>,
                        ops::PnormGradKernel<CPU, double>);
+REGISTER_OP_VERSION(p_norm)
+    .AddCheckpoint(
+        R"ROC(
+        Upgrade p_norm, add 1 attribute [asvector].
+      )ROC",
+        paddle::framework::compatible::OpVersionDesc().NewAttr(
+            "asvector",
+            "Compute as vector when axis is None and input is matrix", false));