Merge pull request #5 from firedrakeproject/fix-literals

Symbolic: Fix handling of Literals.
firedrakeproject · Jul 14, 2020 · 83c0ce0 · 83c0ce0
2 parents cf0a082 + 1e2f80a
commit 83c0ce0
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/loopy/symbolic.py b/loopy/symbolic.py
@@ -361,6 +361,9 @@ def map_type_cast(self, expr, *args, **kwargs):
     def map_resolved_function(self, expr):
         return self.rec(expr.function)
 
+    def map_literal(self, expr):
+        return set()
+
 
 class SubstitutionRuleExpander(IdentityMapper):
     def __init__(self, rules):
@@ -1644,6 +1647,10 @@ def map_remainder(self, expr):
 
         return num.mod_val(denom)
 
+    def map_literal(self, expr):
+        raise TypeError("literal '%s' not supported "
+                        "for as-pwaff evaluation" % expr)
+
 
 def aff_from_expr(space, expr, vars_to_zero=None):
     if vars_to_zero is None: