From 2b53dcb648492462c78dd6bcd3f5e834fac68d80 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Tue, 9 Jul 2013 10:23:07 -0400
Subject: [PATCH] Find potential arg names to guess in array shapes
---
loopy/kernel/creation.py | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/loopy/kernel/creation.py b/loopy/kernel/creation.py
index cf07f969b..39a40728b 100644
--- a/loopy/kernel/creation.py
+++ b/loopy/kernel/creation.py
@@ -426,6 +426,12 @@ def guess_kernel_args_if_requested(domains, instructions, temporary_variables,
all_names.update(get_dependencies(submap(insn.assignee, insn.id)))
all_names.update(get_dependencies(submap(insn.expression, insn.id)))
+ from loopy.kernel.data import ArrayBase
+ for arg in kernel_args:
+ if isinstance(arg, ArrayBase):
+ if isinstance(arg.shape, tuple):
+ all_names.update(get_dependencies(arg.shape))
+
all_params = set()
for dom in domains:
all_params.update(dom.get_var_names(dim_type.param))
--
GitLab