From 42c1d0638d32970a097795232702a7b38c15a86e Mon Sep 17 00:00:00 2001 From: HendrikRiedmann Date: Sat, 12 Sep 2009 18:09:31 -0400 Subject: [PATCH] Fix CSE processing by CCodeMapper. --- pymbolic/mapper/c_code.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pymbolic/mapper/c_code.py b/pymbolic/mapper/c_code.py index 6790b4f..dc36040 100644 --- a/pymbolic/mapper/c_code.py +++ b/pymbolic/mapper/c_code.py @@ -64,7 +64,7 @@ class CCodeMapper(SimplifyingSortingStringifyMapper): def map_common_subexpression(self, expr, enclosing_prec): try: - cse_name = self.cse_to_name[expr] + cse_name = self.cse_to_name[expr.child] except KeyError: from pymbolic.mapper.stringifier import PREC_NONE cse_str = self.rec(expr.child, PREC_NONE) @@ -91,4 +91,6 @@ class CCodeMapper(SimplifyingSortingStringifyMapper): self.cse_to_name[expr.child] = cse_name self.cse_names.add(cse_name) + assert len(self.cse_names) == len(self.cse_to_name) + return cse_name -- GitLab