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

Deal with @-signs of iname-duplicating reductions in more places.

parent 0ec159e1
No related branches found
No related tags found
No related merge requests found
...@@ -260,7 +260,7 @@ class Instruction(Record): ...@@ -260,7 +260,7 @@ class Instruction(Record):
def map_reduction(expr, rec): def map_reduction(expr, rec):
rec(expr.expr) rec(expr.expr)
for iname in expr.inames: for iname in expr.inames:
result.add(iname) result.add(iname.lstrip("@"))
from loopy.symbolic import ReductionCallbackMapper from loopy.symbolic import ReductionCallbackMapper
cb_mapper = ReductionCallbackMapper(map_reduction) cb_mapper = ReductionCallbackMapper(map_reduction)
......
...@@ -546,7 +546,7 @@ class IndexVariableFinder(CombineMapper): ...@@ -546,7 +546,7 @@ class IndexVariableFinder(CombineMapper):
if self.include_reduction_inames: if self.include_reduction_inames:
return result return result
else: else:
return result - set(expr.inames) return result - real_inames
# }}} # }}}
......
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