From 9bb1e1420f2329a996d3db58ad7c68a28477bc8c Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Mon, 15 Jul 2019 10:14:43 +0200
Subject: [PATCH] Add custom_backend to pystencils.show_code
---
pystencils/display_utils.py | 12 +++++++-----
1 file changed, 7 insertions(+), 5 deletions(-)
diff --git a/pystencils/display_utils.py b/pystencils/display_utils.py
index 55a4720..8cdaa48 100644
--- a/pystencils/display_utils.py
+++ b/pystencils/display_utils.py
@@ -1,5 +1,7 @@
-import sympy as sp
from typing import Any, Dict, Optional
+
+import sympy as sp
+
from pystencils.astnodes import KernelFunction
@@ -32,7 +34,7 @@ def highlight_cpp(code: str):
return HTML(highlight(code, CppLexer(), HtmlFormatter()))
-def show_code(ast: KernelFunction):
+def show_code(ast: KernelFunction, custom_backend=None):
"""Returns an object to display generated code (C/C++ or CUDA)
Can either be displayed as HTML in Jupyter notebooks or printed as normal string.
@@ -45,11 +47,11 @@ def show_code(ast: KernelFunction):
self.ast = ast_input
def _repr_html_(self):
- return highlight_cpp(generate_c(self.ast, dialect=dialect)).__html__()
+ return highlight_cpp(generate_c(self.ast, dialect=dialect, custom_backend=custom_backend)).__html__()
def __str__(self):
- return generate_c(self.ast, dialect=dialect)
+ return generate_c(self.ast, dialect=dialect, custom_backend=custom_backend)
def __repr__(self):
- return generate_c(self.ast, dialect=dialect)
+ return generate_c(self.ast, dialect=dialect, custom_backend=custom_backend)
return CodeDisplay(ast)
--
GitLab