From 7fcdb46992e97d635d80430028d24f937f92cc70 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sat, 27 Jan 2018 22:01:43 -0600
Subject: [PATCH] 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