From 23992e8a6e8b69c08edbaf126e57037df40e9728 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Sat, 27 Jan 2018 22:01:14 -0600 Subject: [PATCH 1/2] Allow make_kernel to take a single BasicSet --- loopy/kernel/creation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/loopy/kernel/creation.py b/loopy/kernel/creation.py index 7acb53f8e..4a08c28bd 100644 --- a/loopy/kernel/creation.py +++ b/loopy/kernel/creation.py @@ -1004,7 +1004,7 @@ def _find_existentially_quantified_inames(dom_str): def parse_domains(domains, defines): - if isinstance(domains, str): + if isinstance(domains, (isl.BasicSet, str)): domains = [domains] result = [] -- GitLab From 7fcdb46992e97d635d80430028d24f937f92cc70 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Sat, 27 Jan 2018 22:01:43 -0600 Subject: [PATCH 2/2] Add expect_completion arg to add_and_infer_dtypes, default to False --- loopy/kernel/tools.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/loopy/kernel/tools.py b/loopy/kernel/tools.py index a65e7fb4c..fbc4238c2 100644 --- a/loopy/kernel/tools.py +++ b/loopy/kernel/tools.py @@ -107,7 +107,7 @@ def get_arguments_with_incomplete_dtype(knl): if arg.dtype is None] -def add_and_infer_dtypes(knl, dtype_dict): +def add_and_infer_dtypes(knl, dtype_dict, expect_completion=False): processed_dtype_dict = {} for k, v in six.iteritems(dtype_dict): @@ -119,7 +119,7 @@ def add_and_infer_dtypes(knl, dtype_dict): knl = add_dtypes(knl, processed_dtype_dict) from loopy.type_inference import infer_unknown_types - return infer_unknown_types(knl, expect_completion=True) + return infer_unknown_types(knl, expect_completion=expect_completion) def _add_and_infer_dtypes_overdetermined(knl, dtype_dict): -- GitLab