From 7c27894a8d3332876f4f86dc2c564341dd41557d Mon Sep 17 00:00:00 2001
From: arghdos <arghdos@gmail.com>
Date: Tue, 2 May 2017 13:24:41 -0400
Subject: [PATCH] fix wrong import

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

diff --git a/loopy/auto_test.py b/loopy/auto_test.py
index 537e65fb3..9fefa5e3e 100644
--- a/loopy/auto_test.py
+++ b/loopy/auto_test.py
@@ -403,7 +403,8 @@ def auto_test_vs_ref(
             raise LoopyError("ref_knl and test_knl argument lists disagree at index "
                     "%d (1-based)" % (i+1))
 
-    from loopy.compiled import CompiledKernel, get_highlighted_code
+    from loopy.compiled import CompiledKernel
+    from loopy.execution import get_highlighted_code
 
     if isinstance(op_count, (int, float)):
         warn("op_count should be a list", stacklevel=2)
-- 
GitLab