From 32bb1d31c438e0861a13ed1edf0e4e9a4b79a6ba Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Tue, 6 Oct 2020 14:47:19 +0200
Subject: [PATCH] Add printing for BlockWithOutBraces
---
src/pystencils_autodiff/framework_integration/printer.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/src/pystencils_autodiff/framework_integration/printer.py b/src/pystencils_autodiff/framework_integration/printer.py
index b05d142..edc6edc 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:
--
GitLab