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

Add remove_cses().

parent bb11a67a
No related branches found
No related tags found
No related merge requests found
......@@ -453,6 +453,20 @@ def add_prefetch(kernel, var_name, fetch_dims=[], uni_template=None,
# }}}
def remove_cses(kernel):
from loopy.symbolic import CSECallbackMapper
def map_cse(expr, rec):
return expr.child
new_insns = []
for insn in kernel.instructions:
new_insns.append(
insn.copy(
expression=CSECallbackMapper(map_cse)(insn.expression)))
return kernel.copy(instructions=new_insns)
......
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