From 325ba4dcf72df2be24397def2b51b1f3f67ca7eb Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Mon, 27 Feb 2017 08:53:57 -0600
Subject: [PATCH] Make to_batched recurse on indices (reported by Marmaduke
Woodman)
---
loopy/transform/batch.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/loopy/transform/batch.py b/loopy/transform/batch.py
index 88e3898e2..ea575dd48 100644
--- a/loopy/transform/batch.py
+++ b/loopy/transform/batch.py
@@ -63,7 +63,7 @@ class _BatchVariableChanger(RuleAwareIdentityMapper):
if not self.needs_batch_subscript(expr.aggregate.name):
return super(_BatchVariableChanger, self).map_subscript(expr, expn_state)
- idx = expr.index
+ idx = self.rec(expr.index, expn_state)
if not isinstance(idx, tuple):
idx = (idx,)
--
GitLab