From 92adeceb6812eb462abdeaa6e8bf9a5cd9adf1e7 Mon Sep 17 00:00:00 2001
From: tianshuo78520a <707759223@qq.com>
Date: Mon, 10 May 2021 14:24:00 +0800
Subject: [PATCH] make check_op_desc.py support python3 (#32807)

---
 tools/check_op_desc.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/tools/check_op_desc.py b/tools/check_op_desc.py
index 15e41040121..9e5bd08be69 100644
--- a/tools/check_op_desc.py
+++ b/tools/check_op_desc.py
@@ -14,6 +14,7 @@
 
 import json
 import sys
+import operator
 from paddle.utils import OpLastCheckpointChecker
 from paddle.fluid.core import OpUpdateType
 
@@ -71,7 +72,8 @@ def diff_vars(origin_vars, new_vars):
     vars_name_only_in_new = set(new_vars.keys()) - set(origin_vars.keys())
 
     for var_name in common_vars_name:
-        if cmp(origin_vars.get(var_name), new_vars.get(var_name)) == SAME:
+        if operator.eq(origin_vars.get(var_name),
+                       new_vars.get(var_name)) == SAME:
             continue
         else:
             error, var_error = True, True
@@ -120,7 +122,8 @@ def diff_attr(ori_attrs, new_attrs):
     attrs_only_in_new = set(new_attrs.keys()) - set(ori_attrs.keys())
 
     for attr_name in common_attrs:
-        if cmp(ori_attrs.get(attr_name), new_attrs.get(attr_name)) == SAME:
+        if operator.eq(ori_attrs.get(attr_name),
+                       new_attrs.get(attr_name)) == SAME:
             continue
         else:
             error, attr_error = True, True
@@ -184,7 +187,7 @@ def compare_op_desc(origin_op_desc, new_op_desc):
     new = json.loads(new_op_desc)
     desc_error_message = {}
     version_error_message = {}
-    if cmp(origin_op_desc, new_op_desc) == SAME:
+    if operator.eq(origin_op_desc, new_op_desc) == SAME:
         return desc_error_message, version_error_message
 
     for op_type in origin:
-- 
GitLab