From e2e48066f4813914fedd3ec88990b6b08e9031bf Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Fri, 11 Nov 2011 20:03:00 -0500 Subject: [PATCH] Fix exposure of with_annotation flag. --- loopy/codegen/__init__.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/loopy/codegen/__init__.py b/loopy/codegen/__init__.py index c397e951f..08e3c7a7c 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() -- GitLab