From 6ef2c660cee272af35fceb2310ad34256ce32fef Mon Sep 17 00:00:00 2001
From: Tim Warburton <timwar@caam.rice.edu>
Date: Tue, 8 Nov 2011 23:55:03 -0600
Subject: [PATCH] Add edit_code flag to automated testing harness.
---
loopy/compiled.py | 12 +++++++++---
1 file changed, 9 insertions(+), 3 deletions(-)
diff --git a/loopy/compiled.py b/loopy/compiled.py
index 8baede251..335395db0 100644
--- a/loopy/compiled.py
+++ b/loopy/compiled.py
@@ -22,8 +22,9 @@ class CompiledKernel:
import pyopencl as cl
try:
+ self.cl_program = cl.Program(context, self.code)
self.cl_kernel = getattr(
- cl.Program(context, self.code).build(options=options),
+ self.cl_program.build(options=options),
kernel.name)
except:
print "[Loopy] ----------------------------------------------------"
@@ -225,7 +226,8 @@ def make_args(queue, kernel, seq_input_arrays, parameters):
def auto_test_vs_seq(seq_knl, ctx, kernel_gen, op_count, op_label, parameters,
- print_seq_code=False, print_code=True, warmup_rounds=2, timing_rounds=100):
+ print_seq_code=False, print_code=True, warmup_rounds=2, timing_rounds=100,
+ edit_code=False, dump_binary=False):
from time import time
# {{{ set up CL context for sequential run
@@ -300,13 +302,17 @@ def auto_test_vs_seq(seq_knl, ctx, kernel_gen, op_count, op_label, parameters,
if args is None:
args, output_arrays = make_args(queue, kernel, seq_input_arrays, parameters)
- compiled = CompiledKernel(ctx, kernel)
+ compiled = CompiledKernel(ctx, kernel, edit_code=edit_code)
print "----------------------------------------------------------"
print "Kernel #%d:" % i
print "----------------------------------------------------------"
if print_code:
print_highlighted_code(compiled.code)
print "----------------------------------------------------------"
+ if dump_binary:
+ print type(compiled.cl_program)
+ print compiled.cl_program.binaries[0]
+ print "----------------------------------------------------------"
do_check = True
--
GitLab