From e779b244cf024fca68d39e9e9363b1e236388bc1 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner <inform@tiker.net> Date: Wed, 1 Mar 2017 09:43:25 -0500 Subject: [PATCH] Fix unsubscripted-variable case of to_batched (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 ea575dd48..e7a86300f 100644 --- a/loopy/transform/batch.py +++ b/loopy/transform/batch.py @@ -73,7 +73,7 @@ class _BatchVariableChanger(RuleAwareIdentityMapper): if not self.needs_batch_subscript(expr.name): return super(_BatchVariableChanger, self).map_variable(expr, expn_state) - return expr.aggregate[self.batch_iname_expr] + return expr[self.batch_iname_expr] def _add_unique_dim_name(name, dim_names): -- GitLab