From feb141d85b0b1a950fa97cb5cd2d9b4ae27fe3ea Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Mon, 11 Apr 2016 17:44:56 -0500
Subject: [PATCH] Allow comma-separated keys in add_and_infer_dtypes

---
 loopy/kernel/tools.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/loopy/kernel/tools.py b/loopy/kernel/tools.py
index cb72f283e..d7018a702 100644
--- a/loopy/kernel/tools.py
+++ b/loopy/kernel/tools.py
@@ -102,7 +102,17 @@ def get_arguments_with_incomplete_dtype(knl):
 
 
 def add_and_infer_dtypes(knl, dtype_dict):
-    knl = add_dtypes(knl, dtype_dict)
+    processed_dtype_dict = {}
+
+    for k, v in six.iteritems(dtype_dict):
+        for subkey in k.split(","):
+            subkey = subkey.strip()
+            if subkey:
+                processed_dtype_dict[subkey] = v
+
+    print(processed_dtype_dict)
+
+    knl = add_dtypes(knl, processed_dtype_dict)
 
     from loopy.preprocess import infer_unknown_types
     return infer_unknown_types(knl, expect_completion=True)
-- 
GitLab