From 08c0e925cf77fa0ef038c2ae2ebace1289befd9d Mon Sep 17 00:00:00 2001
From: jdsteve2 <jdsteve2@illinois.edu>
Date: Sun, 22 Mar 2020 15:01:52 -0500
Subject: [PATCH] move code for getting used inames into separate function:
get_used_inames()
---
loopy/transform/iname.py | 25 ++++++++++++++-----------
1 file changed, 14 insertions(+), 11 deletions(-)
diff --git a/loopy/transform/iname.py b/loopy/transform/iname.py
index 96c8252ef..91170647b 100644
--- a/loopy/transform/iname.py
+++ b/loopy/transform/iname.py
@@ -1184,6 +1184,19 @@ def rename_iname(knl, old_iname, new_iname, existing_ok=False, within=None):
# {{{ remove unused inames
+def get_used_inames(knl):
+ import loopy as lp
+ exp_knl = lp.expand_subst(knl)
+
+ used_inames = set()
+ for insn in exp_knl.instructions:
+ used_inames.update(
+ exp_knl.insn_inames(insn.id)
+ | insn.reduction_inames())
+
+ return used_inames
+
+
def remove_unused_inames(knl, inames=None):
"""Delete those among *inames* that are unused, i.e. project them
out of the domain. If these inames pose implicit restrictions on
@@ -1204,17 +1217,7 @@ 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 exp_knl.instructions:
- used_inames.update(
- exp_knl.insn_inames(insn.id)
- | insn.reduction_inames())
-
- unused_inames = inames - used_inames
+ unused_inames = set(inames) - get_used_inames(knl)
# }}}
--
GitLab