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
var(iname) for iname in duplicate_inames
)]
from pymbolic import substitute
new_inner_expr = substitute(rec(expr.child), dict(
(old_iname, var(new_iname))
for old_iname, new_iname in zip(duplicate_inames, new_inames)))
from loopy.symbolic import SubstitutionMapper
from pymbolic.mapper.substitutor import make_subst_func
subst_map = SubstitutionMapper(make_subst_func(
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