From ac32b82b2ccf47299d8a26e7c0654b417cf5c157 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Wed, 12 Oct 2011 08:04:39 -0400
Subject: [PATCH] Highlight source code on output if Pygments is present.
---
MEMO | 3 +++
loopy/compiled.py | 11 ++++++++++-
loopy/schedule.py | 3 +--
test/test_matmul.py | 2 +-
4 files changed, 15 insertions(+), 4 deletions(-)
diff --git a/MEMO b/MEMO
index 9aec2cce9..8eb2fd5ee 100644
--- a/MEMO
+++ b/MEMO
@@ -80,6 +80,9 @@ TODO
- Flag, exploit idempotence
+- Implement insert_parallel_dim_check_points
+ (but first: find a kernel that needs it)
+
Dealt with
^^^^^^^^^^
diff --git a/loopy/compiled.py b/loopy/compiled.py
index ed80d2371..85daa1612 100644
--- a/loopy/compiled.py
+++ b/loopy/compiled.py
@@ -98,7 +98,16 @@ def drive_timing_run(kernel_generator, queue, launch, flop_count=None,
print "SOLUTION #%d" % soln_count
print "-----------------------------------------------"
if print_code:
- print compiled.code
+ try:
+ from pygments import highlight
+ except ImportError:
+ print compiled.code
+ else:
+ from pygments.lexers import CLexer
+ from pygments.formatters import TerminalFormatter
+
+ print highlight(compiled.code, CLexer(), TerminalFormatter())
+
print "-----------------------------------------------"
elapsed = time_run(compiled)
diff --git a/loopy/schedule.py b/loopy/schedule.py
index f761bc1c9..5fef6bcfb 100644
--- a/loopy/schedule.py
+++ b/loopy/schedule.py
@@ -668,8 +668,7 @@ def insert_barriers(kernel, schedule, level=0):
def insert_parallel_dim_check_points(kernel, schedule):
- from warnings import warn
- warn("insert_parallel_dim_check_points is unimplemented")
+ # FIXME: Unimplemented
return kernel
diff --git a/test/test_matmul.py b/test/test_matmul.py
index 418ab1201..6a54e5864 100644
--- a/test/test_matmul.py
+++ b/test/test_matmul.py
@@ -217,7 +217,7 @@ def test_plain_matrix_mul_new_ui(ctx_factory):
outer_tag="g.0", inner_tag="l.1", no_slabs=True)
knl = lp.split_dimension(knl, "j", 16,
outer_tag="g.1", inner_tag="l.0", no_slabs=True)
- knl = lp.split_dimension(knl, "k", 16, no_slabs=True)
+ knl = lp.split_dimension(knl, "k", 16)
knl = lp.realize_cse(knl, "lhsmat", dtype, ["k_inner", "i_inner"])
knl = lp.realize_cse(knl, "rhsmat", dtype, ["j_inner", "k_inner"])
--
GitLab