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