From 2c263f0e95afdefa3904ab032961666202c0fe59 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Mon, 6 Jun 2016 15:17:54 -0500
Subject: [PATCH] Fix per-subkernel grid size finding

---
 loopy/kernel/__init__.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/loopy/kernel/__init__.py b/loopy/kernel/__init__.py
index 3d64d227e..bfd8ee284 100644
--- a/loopy/kernel/__init__.py
+++ b/loopy/kernel/__init__.py
@@ -918,9 +918,7 @@ class LoopKernel(RecordWithoutPickling):
 
         all_inames_by_insns = set()
         for insn_id in insn_ids:
-            insn = self.id_to_insn[insn_id]
-
-            all_inames_by_insns |= self.insn_inames(insn)
+            all_inames_by_insns |= self.insn_inames(insn_id)
 
         if not all_inames_by_insns <= self.all_inames():
             raise RuntimeError("some inames collected from instructions (%s) "
@@ -935,7 +933,7 @@ class LoopKernel(RecordWithoutPickling):
                 GroupIndexTag, LocalIndexTag,
                 AutoLocalIndexTagBase)
 
-        for iname in self.all_inames():
+        for iname in all_inames_by_insns:
             tag = self.iname_to_tag.get(iname)
 
             if isinstance(tag, GroupIndexTag):
-- 
GitLab