From c109ac02db058a17268d8e191f7720cc27f857b5 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Fri, 28 Feb 2020 13:06:00 +0100
Subject: [PATCH] Remove show_code_debug (DebugFrameworkPrinter is a hack!)
---
src/pystencils_autodiff/__init__.py | 5 ++---
src/pystencils_autodiff/framework_integration/printer.py | 3 +--
2 files changed, 3 insertions(+), 5 deletions(-)
diff --git a/src/pystencils_autodiff/__init__.py b/src/pystencils_autodiff/__init__.py
index 83cc034..273c5cd 100644
--- a/src/pystencils_autodiff/__init__.py
+++ b/src/pystencils_autodiff/__init__.py
@@ -8,7 +8,7 @@ from pystencils_autodiff._autodiff import (
AutoDiffAstPair, AutoDiffBoundaryHandling, AutoDiffOp, DiffModes, create_backward_assignments,
get_jacobian_of_assignments)
from pystencils_autodiff.field_tensor_conversion import ArrayWrapper, coerce_to_field
-from pystencils_autodiff.framework_integration.printer import show_code, show_code_debug
+from pystencils_autodiff.framework_integration.printer import show_code
__all__ = ['backends',
'AdjointField',
@@ -21,8 +21,7 @@ __all__ = ['backends',
'AutoDiffBoundaryHandling',
'ArrayWrapper',
'coerce_to_field',
- 'show_code',
- 'show_code_debug']
+ 'show_code']
sys.modules['pystencils.autodiff'] = pystencils_autodiff
sys.modules['pystencils.autodiff.backends'] = pystencils_autodiff.backends
diff --git a/src/pystencils_autodiff/framework_integration/printer.py b/src/pystencils_autodiff/framework_integration/printer.py
index 2627963..b05d142 100644
--- a/src/pystencils_autodiff/framework_integration/printer.py
+++ b/src/pystencils_autodiff/framework_integration/printer.py
@@ -119,7 +119,7 @@ class FrameworkIntegrationPrinter(pystencils.backends.cbackend.CBackend):
def _print_Timeloop(self, node):
children_string = '\n '.join(self._print(c) for c in node.children)
- return f"""for( {node.loop_symbol.dtype} {node.loop_symbol}={node.loop_start}; {node.loop_symbol}<= {node.loop_end} ; {node.loop_symbol}+= {node.loop_increment} ) {{
+ return f"""for( {node.loop_symbol.dtype} {node.loop_symbol}={node.loop_start}; {node.loop_symbol}<= {node.loop_end} ; {node.loop_symbol} += {node.loop_increment} ) {{
{children_string}
}}""" # noqa
@@ -171,4 +171,3 @@ class DebugFrameworkPrinter(FrameworkIntegrationPrinter):
show_code = functools.partial(pystencils.show_code, custom_backend=FrameworkIntegrationPrinter())
-show_code_debug = functools.partial(pystencils.show_code, custom_backend=DebugFrameworkPrinter())
--
GitLab