diff --git a/src/pystencils_autodiff/framework_integration/printer.py b/src/pystencils_autodiff/framework_integration/printer.py index b05d142decfab2ac04babca117910556bbde8a4a..edc6edc8e50b7c73af9b42c1dfa813545f7478d0 100644 --- a/src/pystencils_autodiff/framework_integration/printer.py +++ b/src/pystencils_autodiff/framework_integration/printer.py @@ -33,6 +33,10 @@ class FrameworkIntegrationPrinter(pystencils.backends.cbackend.CBackend): else: return super()._print(node) + def _print_BlockWithoutBraces(self, node): + block_contents = "\n".join([self._print(child) for child in node.args]) + return "\n%s\n" % (''.join(block_contents.splitlines(True))) + def _print_WrapperFunction(self, node): super_result = super()._print_KernelFunction(node) if self._signatureOnly: