From 9b6b7dccfc16451fa3799359ccbdc20efcfd2241 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Fri, 12 Feb 2016 10:19:50 -0600
Subject: [PATCH] Consider substitution rules in remove_unused_arguments
---
loopy/transform/data.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/loopy/transform/data.py b/loopy/transform/data.py
index f510af598..643326379 100644
--- a/loopy/transform/data.py
+++ b/loopy/transform/data.py
@@ -346,8 +346,11 @@ def set_array_dim_names(kernel, ary_names, dim_names):
def remove_unused_arguments(knl):
new_args = []
+ import loopy as lp
+ exp_knl = lp.expand_subst(knl)
+
refd_vars = set(knl.all_params())
- for insn in knl.instructions:
+ for insn in exp_knl.instructions:
refd_vars.update(insn.dependency_names())
for arg in knl.args:
--
GitLab