diff --git a/loopy/preprocess.py b/loopy/preprocess.py index 583dd9116b5100b35978c85ca40328ac268b8aba..6800a10c66d2fdeb2f1882332ac198f082b07f63 100644 --- a/loopy/preprocess.py +++ b/loopy/preprocess.py @@ -43,7 +43,7 @@ def _infer_var_type(kernel, var_name, type_inf_mapper, subst_expander): dtypes = [] from loopy.codegen.expression import DependencyTypeInferenceFailure - for writer_insn_id in kernel.writer_map()[var_name]: + for writer_insn_id in kernel.writer_map().get(var_name, []): expr = subst_expander( kernel.id_to_insn[writer_insn_id].expression, insn_id=writer_insn_id)