From d41e388262b2a5b07424a634efc6274f1aa17523 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Fri, 31 Aug 2012 01:06:15 -0400
Subject: [PATCH] Allow passing options to auto_test_vs_ref().
---
loopy/compiled.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/loopy/compiled.py b/loopy/compiled.py
index 9b62643e4..8844ffe36 100644
--- a/loopy/compiled.py
+++ b/loopy/compiled.py
@@ -444,6 +444,7 @@ def _default_check_result(result, ref_result):
def auto_test_vs_ref(ref_knl, ctx, kernel_gen, op_count=[], op_label=[], parameters={},
print_ref_code=False, print_code=True, warmup_rounds=2,
edit_code=False, dump_binary=False, codegen_kwargs={},
+ options=[],
fills_entire_output=True, check_result=None):
"""Compare results of `ref_knl` to the kernels generated by the generator
`kernel_gen`.
@@ -518,6 +519,7 @@ def auto_test_vs_ref(ref_knl, ctx, kernel_gen, op_count=[], op_label=[], paramet
break
ref_compiled = CompiledKernel(ref_ctx, ref_sched_kernel,
+ options=options,
codegen_kwargs=codegen_kwargs)
if print_ref_code:
print 75*"-"
@@ -571,6 +573,7 @@ def auto_test_vs_ref(ref_knl, ctx, kernel_gen, op_count=[], op_label=[], paramet
fill_value=fill_value)
compiled = CompiledKernel(ctx, kernel, edit_code=edit_code,
+ options=options,
codegen_kwargs=codegen_kwargs)
print 75*"-"
--
GitLab