From 5ced8e8e207bd7938d4c52a40d548ec29e8d5d28 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Wed, 27 Dec 2017 17:30:46 -0600
Subject: [PATCH] Set target on dtypes in type inference
---
loopy/type_inference.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/loopy/type_inference.py b/loopy/type_inference.py
index 6ffc1dff5..0f58ae64e 100644
--- a/loopy/type_inference.py
+++ b/loopy/type_inference.py
@@ -588,6 +588,9 @@ def infer_unknown_types(kernel, expect_completion=False):
failed = not result
if not failed:
new_dtype, = result
+ if new_dtype.target is None:
+ new_dtype = new_dtype.with_target(kernel.target)
+
debug(" success: %s", new_dtype)
if new_dtype != item.dtype:
debug(" changed from: %s", item.dtype)
--
GitLab