From 6584d232d9ea05c6472263bc187e48a8ce14dedb Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Fri, 12 Feb 2016 10:37:50 -0600 Subject: [PATCH] Consider substitution rules and reduction inames in remove_unused_inames --- loopy/transform/iname.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/loopy/transform/iname.py b/loopy/transform/iname.py index fb22df37a..8cf16bfd3 100644 --- a/loopy/transform/iname.py +++ b/loopy/transform/iname.py @@ -987,10 +987,15 @@ def remove_unused_inames(knl, inames=None): # {{{ check which inames are unused + import loopy as lp + exp_knl = lp.expand_subst(knl) + inames = set(inames) used_inames = set() - for insn in knl.instructions: - used_inames.update(knl.insn_inames(insn.id)) + for insn in exp_knl.instructions: + used_inames.update( + exp_knl.insn_inames(insn.id) + | insn.reduction_inames()) unused_inames = inames - used_inames -- GitLab