PaddlePaddle
diff --git a/‎paddle/phi/kernels/funcs/compare_functors.h
Copy file name to clipboard
+3-10Lines changed: 3 additions & 10 deletions b/‎paddle/phi/kernels/funcs/compare_functors.h
Copy file name to clipboard
+3-10Lines changed: 3 additions & 10 deletions
@@ -33,11 +33,7 @@ namespace funcs {
                                const phi::dtype::complex<T> b) const {      \
       if (isnan(a.real) || isnan(a.imag) || isnan(b.real) || isnan(b.imag)) \
         return false;                                                       \
-      T ar = a.real;                                                        \
-      T br = b.real;                                                        \
-      T ai = a.imag;                                                        \
-      T bi = b.imag;                                                        \
-      return (ar op br) || (ar == br && ai op bi);                          \
+      return (a.real op b.real) || (a.real == b.real && a.imag op b.imag);  \
     }                                                                       \
   };
 
@@ -48,11 +44,8 @@ namespace funcs {
                                const phi::dtype::complex<T> b) const {      \
       if (isnan(a.real) || isnan(a.imag) || isnan(b.real) || isnan(b.imag)) \
         return false;                                                       \
-      T ar = a.real;                                                        \
-      T br = b.real;                                                        \
-      T ai = a.imag;                                                        \
-      T bi = b.imag;                                                        \
-      return (ar op br) || (ar == br && ai op_equal bi);                    \
+      return (a.real op b.real) ||                                          \
+             (a.real == b.real && a.imag op_equal b.imag);                  \
     }                                                                       \
   };