diff --git a/loopy/preprocess.py b/loopy/preprocess.py index 7b05efd0b0639d6fb098bbc92a99ea29e323a7b6..812f6d265dabf56441d2d333ab11acc7f43ef8d1 100644 --- a/loopy/preprocess.py +++ b/loopy/preprocess.py @@ -2404,8 +2404,7 @@ def specializing_incomplete_callables(kernel): # only trying to specialize the functions which are not ready # for codegen type_inf_mapper(expr) - inferred_functions = {**inferred_functions, - **type_inf_mapper.specialized_functions} + inferred_functions.update(type_inf_mapper.specialized_functions) elif isinstance(insn, (_DataObliviousInstruction)): pass