fix: handle complex type correctly in greater equal in paddle frontend (

#28701)
ivy-llc · Apr 2, 2024 · b04c444 · b04c444
1 parent 619cce9
commit b04c444
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/ivy/functional/backends/paddle/elementwise.py b/ivy/functional/backends/paddle/elementwise.py
@@ -482,12 +482,16 @@ def greater_equal(
     *,
     out: Optional[paddle.Tensor] = None,
 ) -> paddle.Tensor:
-    x1, x2, ret_dtype = _elementwise_helper(x1, x2)
+    x1, x2, _ = _elementwise_helper(x1, x2)
     if isinstance(x1, paddle.Tensor) and isinstance(x2, paddle.Tensor):
         if paddle.is_complex(x1) and paddle.is_complex(x2):
-            real = paddle.greater_equal(x1.real(), x2.real())
-            imag = paddle.greater_equal(x1.imag(), x2.imag())
-            return paddle.logical_and(real, imag)
+            real_greater_equal = paddle.real(x1) >= paddle.real(x2)
+            real_equal = paddle.real(x1) == paddle.real(x2)
+            imag_greater_equal = paddle.imag(x1) >= paddle.imag(x2)
+            return paddle.logical_or(
+                real_greater_equal, paddle.logical_and(real_equal, imag_greater_equal)
+            )
+
     return paddle.greater_equal(x1, x2)