diff --git a/loopy/type_inference.py b/loopy/type_inference.py index 2d35d7cfa3412da58624270bd3df9ebf67daa9e9..3128a1d529cce5070356162a2b4ec0372e0b4ad6 100644 --- a/loopy/type_inference.py +++ b/loopy/type_inference.py @@ -448,7 +448,7 @@ class TypeInferenceMapper(CombineMapper): def _infer_var_type(kernel, var_name, type_inf_mapper, subst_expander): if var_name in kernel.all_params(): - return [kernel.index_dtype], [] + return [kernel.index_dtype], [], {} from functools import partial debug = partial(_debug, kernel) @@ -475,7 +475,7 @@ def _infer_var_type(kernel, var_name, type_inf_mapper, subst_expander): dtype_sets.append(result) if not dtype_sets: - return None, type_inf_mapper.symbols_with_unknown_types, {} + return None, type_inf_mapper.symbols_with_unknown_types, None result = type_inf_mapper.combine(dtype_sets)