FlattenMapper: claim everything is an integer

inducer · inducer · commit f6da4582ef08 · 2024-11-07T10:39:01.000-06:00
(for now, matches existing behavior, but is clearly incorrect)
diff --git a/loopy/symbolic.py b/loopy/symbolic.py
@@ -205,7 +205,9 @@ def map_resolved_function(self, expr, *args, **kwargs):
 
 
 class FlattenMapper(FlattenMapperBase, IdentityMapperMixin):
-    pass
+    # FIXME: Lies! This needs to be made precise.
+    def is_expr_integer_valued(self, expr: ExpressionT) -> bool:
+        return True
 
 
 def flatten(expr: ArithmeticOrExpressionT) -> ArithmeticOrExpressionT:
