diff --git a/pystencils_benchmark/benchmark.py b/pystencils_benchmark/benchmark.py index 0ee2218a965c138dac5fa77f5be0b2043a91510f..df6bd9b6d0c754a76ca6a44c38c1ffc2299d0505 100644 --- a/pystencils_benchmark/benchmark.py +++ b/pystencils_benchmark/benchmark.py @@ -23,7 +23,9 @@ def generate_benchmark(kernel_asts: Union[KernelFunction, List[KernelFunction]], path: Path = None, *, compiler: Compiler = Compiler.GCC, - likwid: bool = False) -> None: + timing: bool = True, + likwid: bool = False + ) -> None: if path is None: path = Path('.') else: @@ -48,7 +50,7 @@ def generate_benchmark(kernel_asts: Union[KernelFunction, List[KernelFunction]], f.write(source) with open(src_path / 'main.c', 'w+') as f: - f.write(kernel_main(kernel_asts, likwid=likwid)) + f.write(kernel_main(kernel_asts, timing=timing, likwid=likwid)) copy_static_files(path) compiler_toolchain(path, compiler, likwid)