diff --git a/loopy/compiled.py b/loopy/compiled.py index 28481f7a8c6a2129256ac5f34ba543fec24cba70..fe8435be7a5fcdbbf26c7f888b53fffe012d6cc5 100644 --- a/loopy/compiled.py +++ b/loopy/compiled.py @@ -650,8 +650,10 @@ def generate_invoker(kernel, impl_arg_info, flags): # }}} - if flags.print_wrapper: + if flags.print_wrapper == "hl": print get_highlighted_python_code(gen.get()) + elif flags.print_wrapper: + print gen.get() return gen.get_function()