From 1f600d72a7f54a672c1c7c1fda1ad823c0083b1c Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sun, 15 Jul 2012 16:27:07 -0400
Subject: [PATCH] Take args and tempvars into account when building
 seen_dtypes.

---
 loopy/codegen/__init__.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/loopy/codegen/__init__.py b/loopy/codegen/__init__.py
index b969d9896..ac169d5d6 100644
--- a/loopy/codegen/__init__.py
+++ b/loopy/codegen/__init__.py
@@ -336,6 +336,11 @@ def generate_code(kernel, with_annotation=False,
 
     # {{{ handle preambles
 
+    for arg in kernel.args:
+        seen_dtypes.add(arg.dtype)
+    for tv in kernel.temporary_variables:
+        seen_dtypes.add(tv.dtype)
+
     preambles = kernel.preambles[:]
     for prea_gen in kernel.preamble_generators:
         preambles.extend(prea_gen(seen_dtypes, seen_functions))
-- 
GitLab