diff --git a/loopy/codegen/__init__.py b/loopy/codegen/__init__.py index b969d989682b1cdac32e77625f9f73f7facd4e10..ac169d5d692e02d3885b493bd94b413580bca2ea 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))