From 69dd161217d90c51473ba86704d64384756225ce Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Fri, 3 Jun 2016 17:30:55 -0500 Subject: [PATCH] Fix AccessRangeMapper's treatment of linear subscripts --- loopy/symbolic.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/loopy/symbolic.py b/loopy/symbolic.py index a83db407f..cef32851f 100644 --- a/loopy/symbolic.py +++ b/loopy/symbolic.py @@ -1287,7 +1287,10 @@ class AccessRangeMapper(WalkMapper): self.access_range = self.access_range | access_range def map_linear_subscript(self, expr, inames): - self.bad_subscripts.append(expr) + self.rec(expr.index, inames) + + if expr.aggregate.name == self.arg_name: + self.bad_subscripts.append(expr) def map_reduction(self, expr, inames): return WalkMapper.map_reduction(self, expr, inames | set(expr.inames)) -- GitLab