Numba fallback complex erf

Author: ricardoV94

pytensor/link/numba/dispatch/scalar.py

@@ -278,7 +278,11 @@ def logp1mexp(x):
 
 
 @register_funcify_and_cache_key(Erf)
-def numba_funcify_Erf(op, **kwargs):
+def numba_funcify_Erf(op, node, **kwargs):
+    if node.inputs[0].type.dtype.startswith("complex"):
+        # Complex not supported by numba
+        return numba_funcify_ScalarOp(op, node=node, **kwargs)
+
     @numba_basic.numba_njit
     def erf(x):
         return math.erf(x)

tests/link/numba/test_scalar.py

@@ -216,3 +216,14 @@ def impl(self, x):
             [g],
             [np.array(5, dtype="int64")],
         )
+
+
+def test_erf_complex():
+    x = pt.scalar("x", dtype="complex128")
+    g = pt.erf(x)
+
+    compare_numba_and_py(
+        [x],
+        [g],
+        [np.array(0.5 + 1j, dtype="complex128")],
+    )