Skip to content
Snippets Groups Projects
Commit 0026eb8b authored by Tim Warburton's avatar Tim Warburton
Browse files

Use loopy's own SubstitutionMapper (not pymbolic's substitute) in realize_cse().

parent cc5a489a
No related branches found
No related tags found
No related merge requests found
...@@ -320,10 +320,13 @@ def realize_cse(kernel, cse_tag, dtype, duplicate_inames=[], parallel_inames=Non ...@@ -320,10 +320,13 @@ def realize_cse(kernel, cse_tag, dtype, duplicate_inames=[], parallel_inames=Non
var(iname) for iname in duplicate_inames var(iname) for iname in duplicate_inames
)] )]
from pymbolic import substitute from loopy.symbolic import SubstitutionMapper
new_inner_expr = substitute(rec(expr.child), dict( from pymbolic.mapper.substitutor import make_subst_func
(old_iname, var(new_iname)) subst_map = SubstitutionMapper(make_subst_func(
for old_iname, new_iname in zip(duplicate_inames, new_inames))) dict(
(old_iname, var(new_iname))
for old_iname, new_iname in zip(duplicate_inames, new_inames))))
new_inner_expr = subst_map(rec(expr.child))
# }}} # }}}
......
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