From 4eb67253606711300a6109d5eb6520444c252244 Mon Sep 17 00:00:00 2001
From: Isuru Fernando <isuruf@gmail.com>
Date: Sun, 24 Feb 2019 16:07:40 -0600
Subject: [PATCH 1/2] Rewrite complex coefficients that cannot be represented
 using single precision

Loopy stops with an exception in this case, asking the user to explicitly cast to
np.complex64/128
---
 sumpy/codegen.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/sumpy/codegen.py b/sumpy/codegen.py
index 61ed7301..798365fc 100644
--- a/sumpy/codegen.py
+++ b/sumpy/codegen.py
@@ -581,6 +581,28 @@ class BigIntegerKiller(CSECachingMapperMixin, IdentityMapper):
 # }}}
 
 
+# {{{ convert 123000000j to 123000000 * 1j
+
+class ComplexRewriter(CSECachingMapperMixin, IdentityMapper):
+
+    def __init__(self, float_type=np.float32):
+        IdentityMapper.__init__(self)
+        self.float_type = float_type
+
+    def map_constant(self, expr):
+        """Convert complex values not within complex64 to a product for loopy
+        """
+        if not isinstance(expr, complex):
+            return IdentityMapper.map_constant(self, expr)
+
+        if complex(self.float_type(expr.imag)) == expr.imag:
+            return IdentityMapper.map_constant(self, expr)
+
+        return expr.real + prim.Product((expr.imag, 1j))
+
+    map_common_subexpression_uncached = IdentityMapper.map_common_subexpression
+
+
 # {{{ vector component rewriter
 
 INDEXED_VAR_RE = re.compile("^([a-zA-Z_]+)([0-9]+)$")
@@ -684,6 +706,7 @@ def to_loopy_insns(assignments, vector_names=set(), pymbolic_expr_maps=[],
     ssg = SumSignGrouper()
     fck = FractionKiller()
     bik = BigIntegerKiller()
+    cmr  = ComplexRewriter()
 
     def convert_expr(name, expr):
         logger.debug("generate expression for: %s" % name)
@@ -694,6 +717,7 @@ def to_loopy_insns(assignments, vector_names=set(), pymbolic_expr_maps=[],
         expr = fck(expr)
         expr = ssg(expr)
         expr = bik(expr)
+        expr = cmr(expr)
         #expr = cse_tag(expr)
         for m in pymbolic_expr_maps:
             expr = m(expr)
-- 
GitLab


From 1bfcccebaeb7d836c363b13a3501bf75c4e09090 Mon Sep 17 00:00:00 2001
From: Isuru Fernando <isuruf@gmail.com>
Date: Sun, 24 Feb 2019 16:59:07 -0600
Subject: [PATCH 2/2] formatting fixes

---
 sumpy/codegen.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sumpy/codegen.py b/sumpy/codegen.py
index 798365fc..bef7fc6b 100644
--- a/sumpy/codegen.py
+++ b/sumpy/codegen.py
@@ -706,7 +706,7 @@ def to_loopy_insns(assignments, vector_names=set(), pymbolic_expr_maps=[],
     ssg = SumSignGrouper()
     fck = FractionKiller()
     bik = BigIntegerKiller()
-    cmr  = ComplexRewriter()
+    cmr = ComplexRewriter()
 
     def convert_expr(name, expr):
         logger.debug("generate expression for: %s" % name)
-- 
GitLab