[API Compatibility] Fix tensor __eq__ and __ne__ for unsupported type (#76118)

* [API Compatibility] Support tensor equals None usage

* update test case

* support __ne__ for unsupported type

* support pir.Value

* update

* update

* update

---------

Co-authored-by: SigureMo <[email protected]>

Author: huangjiyi

paddle/fluid/pybind/eager_math_op_patch.cc

@@ -2204,8 +2204,16 @@ static PyObject* tensor__ne__method(TensorObject* self,
       other_tensor = paddle::empty({}, phi::DataType::FLOAT32, place);
       InitTensorWithNumpyValue(numpy_value, place, &other_tensor);
     } else {
-      paddle::experimental::Scalar value =
-          CastPyArg2Scalar(other_obj, "__ne__", 0);
+      paddle::experimental::Scalar value;
+
+      // return True if other_obj is unsupported type
+      try {
+        value = CastPyArg2Scalar(other_obj, "__ne__", 0);
+      } catch (const ::common::enforce::EnforceNotMet& e) {
+        Py_INCREF(Py_True);
+        return Py_True;
+      }
+
       if (PyComplex_Check(other_obj)) {
         eager_gil_scoped_release guard;
         other_tensor =
@@ -2297,8 +2305,16 @@ static PyObject* tensor__eq__method(TensorObject* self,
       other_tensor = paddle::empty({}, phi::DataType::FLOAT32, place);
       InitTensorWithNumpyValue(numpy_value, place, &other_tensor);
     } else {
-      paddle::experimental::Scalar value =
-          CastPyArg2Scalar(other_obj, "__eq__", 0);
+      paddle::experimental::Scalar value;
+
+      // return False if other_obj is unsupported type
+      try {
+        value = CastPyArg2Scalar(other_obj, "__eq__", 0);
+      } catch (const ::common::enforce::EnforceNotMet& e) {
+        Py_INCREF(Py_False);
+        return Py_False;
+      }
+
       if (PyComplex_Check(other_obj)) {
         eager_gil_scoped_release guard;
         other_tensor =

python/paddle/pir/math_op_patch.py

@@ -528,6 +528,13 @@ def __impl__(self, other_var):
                 # but only +, -, *, / can use this method
                 if scalar_method is not None:
                     return scalar_method(self, other_var)
+            elif other_var is None:
+                if method_name == "__eq__":
+                    return False
+                elif method_name == "__ne__":
+                    return True
+                else:
+                    pass
             else:
                 # do nothing
                 pass

test/legacy_test/test_eager_tensor.py

@@ -1592,6 +1592,17 @@ def test_tensor__format__(self):
         paddle_scalar = paddle.uniform([], min=-100, max=100)
         self.assertRaises(ValueError, paddle_scalar.__format__, "3d")
 
+    def test_tensor_eq_unsupported_type(self):
+        a = paddle.empty([2])
+
+        # Compare with None
+        self.assertFalse(a == None)  # noqa: E711
+        self.assertTrue(a != None)  # noqa: E711
+
+        # Compare with other obj
+        self.assertFalse(a == object())
+        self.assertTrue(a != object())
+
 
 class TestEagerTensorSetitem(unittest.TestCase):
     def func_setUp(self):

test/legacy_test/test_math_op_patch_pir.py

@@ -451,6 +451,8 @@ def _test(x_np, y_np, input_dtype):
                     e = x != y
                     f = x.not_equal(y)
                     g = x.__ne__(y)
+                    self.assertFalse(x == None)  # noqa: E711
+                    self.assertTrue(x != None)  # noqa: E711
                     (e_np, f_np, g_np) = exe.run(
                         main_program,
                         feed={"x": x_np, "y": y_np},