diff --git a/loopy/compiled.py b/loopy/compiled.py index 9b62643e479e685a900c44ddb0cd8ff53a9d3b7e..8844ffe36154d9e0800afd4b436a338a0284a29c 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*"-"