diff --git a/src/pystencils/codegen/driver.py b/src/pystencils/codegen/driver.py index 74a07b902a6494d0c528602326286a2aaab01cc5..59e3139133ab76097bd1e8febdae2ffb91eea41f 100644 --- a/src/pystencils/codegen/driver.py +++ b/src/pystencils/codegen/driver.py @@ -320,8 +320,7 @@ class DefaultKernelCreationDriver: append_ast = [PsAssignment(ptr_access, write_back_ptr)] # modify AST - kernel_ast.statements = prepend_ast + kernel_ast.statements - kernel_ast.statements += append_ast + kernel_ast.statements = prepend_ast + kernel_ast.statements + append_ast def _transform_for_cpu(self, kernel_ast: PsBlock) -> PsBlock: canonicalize = CanonicalizeSymbols(self._ctx, True)