diff --git a/pystencils_benchmark/benchmark.py b/pystencils_benchmark/benchmark.py index 41d7410f50c77bb0c838238ff4f5e0c10a85af3c..0ee2218a965c138dac5fa77f5be0b2043a91510f 100644 --- a/pystencils_benchmark/benchmark.py +++ b/pystencils_benchmark/benchmark.py @@ -95,6 +95,7 @@ def kernel_main(kernels_ast: List[KernelFunction], *, Args: kernels_ast: A list of the pystencils AST object as returned by create_kernel for benchmarking timing: add timing output to the code, prints time per iteration to stdout + likwid: add likwid marker to the code Returns: C code as string """ @@ -134,7 +135,8 @@ def kernel_main(kernels_ast: List[KernelFunction], *, size = dim1_size * padding_bytes + np.prod(field.shape) * size_data_type assert align % np_dtype.itemsize == 0 - offset = ((dim0_size + padding_elements + ghost_layers) % kernel.instruction_set['width']) * size_data_type + offset = ((dim0_size + padding_elements + ghost_layers) % + kernel.instruction_set['width']) * size_data_type else: size = elements * size_data_type offset = 0