From b37aaaf7b80990c361726abb67dff977b24eed60 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Thu, 3 Nov 2011 01:11:31 -0400
Subject: [PATCH] Deal with @-signs of iname-duplicating reductions in more
places.
---
loopy/kernel.py | 2 +-
loopy/symbolic.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/loopy/kernel.py b/loopy/kernel.py
index 25efd65cc..1fbea5493 100644
--- a/loopy/kernel.py
+++ b/loopy/kernel.py
@@ -260,7 +260,7 @@ class Instruction(Record):
def map_reduction(expr, rec):
rec(expr.expr)
for iname in expr.inames:
- result.add(iname)
+ result.add(iname.lstrip("@"))
from loopy.symbolic import ReductionCallbackMapper
cb_mapper = ReductionCallbackMapper(map_reduction)
diff --git a/loopy/symbolic.py b/loopy/symbolic.py
index b4f5871bb..210917484 100644
--- a/loopy/symbolic.py
+++ b/loopy/symbolic.py
@@ -546,7 +546,7 @@ class IndexVariableFinder(CombineMapper):
if self.include_reduction_inames:
return result
else:
- return result - set(expr.inames)
+ return result - real_inames
# }}}
--
GitLab