From aff0c6b877a1abc11043587bd31fd9c2d6091ae2 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Thu, 26 Sep 2019 16:47:39 +0200
Subject: [PATCH] Let show_code also work for KernelWrapper
---
pystencils/display_utils.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/pystencils/display_utils.py b/pystencils/display_utils.py
index 8cdaa482..638d1290 100644
--- a/pystencils/display_utils.py
+++ b/pystencils/display_utils.py
@@ -3,6 +3,7 @@ from typing import Any, Dict, Optional
import sympy as sp
from pystencils.astnodes import KernelFunction
+from pystencils.kernel_wrapper import KernelWrapper
def to_dot(expr: sp.Expr, graph_style: Optional[Dict[str, Any]] = None, short=True):
@@ -40,6 +41,10 @@ def show_code(ast: KernelFunction, custom_backend=None):
Can either be displayed as HTML in Jupyter notebooks or printed as normal string.
"""
from pystencils.backends.cbackend import generate_c
+
+ if isinstance(ast, KernelWrapper):
+ ast = ast.ast
+
dialect = 'cuda' if ast.backend == 'gpucuda' else 'c'
class CodeDisplay:
--
GitLab