diff --git a/loopy/codegen/__init__.py b/loopy/codegen/__init__.py index c397e951f8f35c06f3d4c1f1598c0d4cbe0583de..08e3c7a7c75cc169ac0f2f3090aab9244b27db43 100644 --- a/loopy/codegen/__init__.py +++ b/loopy/codegen/__init__.py @@ -186,9 +186,8 @@ def generate_code(kernel, with_annotation=False): from loopy.symbolic import LoopyCCodeMapper, pw_aff_to_expr - ccm = LoopyCCodeMapper(kernel).copy_and_assign_many( - make_initial_assignments(kernel), - with_annotation=with_annotation) + ccm = (LoopyCCodeMapper(kernel, with_annotation=with_annotation) + .copy_and_assign_many(make_initial_assignments(kernel))) mod = Module()