Skip to content
Snippets Groups Projects
Commit d41e3882 authored by Andreas Klöckner's avatar Andreas Klöckner
Browse files

Allow passing options to auto_test_vs_ref().

parent c17df444
No related branches found
No related tags found
No related merge requests found
......@@ -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*"-"
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment