Fix complex init in scalarmath

lysnikolaou · Jul 7, 2023 · c2fb01c · c2fb01c
1 parent 8437a56
commit c2fb01c
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/numpy/core/src/umath/scalarmath.c.src b/numpy/core/src/umath/scalarmath.c.src
@@ -1625,13 +1625,31 @@ static PyObject *
  *          Byte, UByte, Short, UShort, Int, UInt,
  *          Long, ULong, LongLong, ULongLong#
  *
+ * #ONAME = (BYTE, UBYTE, SHORT, USHORT, INT, UINT,
+ *              LONG, ULONG, LONGLONG, ULONGLONG,
+ *              HALF, FLOAT, DOUBLE, LONGDOUBLE,
+ *              CFLOAT, CDOUBLE, CLONGDOUBLE)*2,
+ *          BYTE, UBYTE, SHORT, USHORT, INT, UINT,
+ *          LONG, ULONG, LONGLONG, ULONGLONG,
+ *          HALF, FLOAT, DOUBLE, LONGDOUBLE,
+ *          FLOAT, DOUBLE, LONGDOUBLE,
+ *
+ *          BYTE, UBYTE, SHORT, USHORT, INT, UINT,
+ *          LONG, ULONG, LONGLONG, ULONGLONG#
+ *
+ * #IS_COMPLEX = 0*14, 1*3, 0*14, 1*3, 0*27#
+ *
  * #oper = negative*17, positive*17, absolute*17, invert*10#
  */
 static PyObject *
 @name@_@oper@(PyObject *a)
 {
     @type@ val;
+#if @IS_COMPLEX@ == 1
+    @otype@ out = NPY_@ONAME@_INIT(0.0, 0.0);
+#else
     @otype@ out;
+#endif
     PyObject *ret;