From 539ef1d16f26361408139ee49283a650d8cecc2d Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Tue, 18 Jun 2013 13:05:45 -0400
Subject: [PATCH] Allow True,False,'hl' for print_wrapper

---
 loopy/compiled.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/loopy/compiled.py b/loopy/compiled.py
index 28481f7a8..fe8435be7 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()
 
-- 
GitLab