Skip to content
Snippets Groups Projects
Commit d662d959 authored by Andreas Klöckner's avatar Andreas Klöckner
Browse files

Merge branch 'master' of hendrikr@hpcgeek:src/pymbolic

parents 9f33b6ee 42c1d063
No related branches found
No related tags found
No related merge requests found
...@@ -55,7 +55,7 @@ class CCodeMapper(SimplifyingSortingStringifyMapper): ...@@ -55,7 +55,7 @@ class CCodeMapper(SimplifyingSortingStringifyMapper):
def map_common_subexpression(self, expr, enclosing_prec): def map_common_subexpression(self, expr, enclosing_prec):
try: try:
cse_name = self.cse_to_name[expr] cse_name = self.cse_to_name[expr.child]
except KeyError: except KeyError:
from pymbolic.mapper.stringifier import PREC_NONE from pymbolic.mapper.stringifier import PREC_NONE
cse_str = self.rec(expr.child, PREC_NONE) cse_str = self.rec(expr.child, PREC_NONE)
...@@ -82,6 +82,8 @@ class CCodeMapper(SimplifyingSortingStringifyMapper): ...@@ -82,6 +82,8 @@ class CCodeMapper(SimplifyingSortingStringifyMapper):
self.cse_to_name[expr.child] = cse_name self.cse_to_name[expr.child] = cse_name
self.cse_names.add(cse_name) self.cse_names.add(cse_name)
assert len(self.cse_names) == len(self.cse_to_name)
return cse_name return cse_name
def map_if_positive(self, expr, enclosing_prec): def map_if_positive(self, expr, enclosing_prec):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment