From a1f16afaf280f9f88eb5fc65a9afb09d9b50d37e Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <kloeckner@teramite.rice.edu>
Date: Wed, 24 Sep 2008 11:35:14 -0500
Subject: [PATCH] Make FlattenMapper recurse properly.
---
src/mapper/flattener.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/mapper/flattener.py b/src/mapper/flattener.py
index 41eaa40..ee05303 100644
--- a/src/mapper/flattener.py
+++ b/src/mapper/flattener.py
@@ -6,11 +6,11 @@ from pymbolic.mapper import IdentityMapper
class FlattenMapper(IdentityMapper):
def map_sum(self, expr):
from pymbolic.primitives import flattened_sum
- return flattened_sum(expr.children)
+ return flattened_sum(self.rec(ch) for ch in expr.children)
def map_product(self, expr):
from pymbolic.primitives import flattened_product
- return flattened_product(expr.children)
+ return flattened_product(self.rec(ch) for ch in expr.children)
def handle_unsupported_expression(self, expr):
return expr
--
GitLab