From aa913014aae9b1fc3872de54f3c342ab65ca465a Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Mon, 3 Jul 2017 20:16:41 -0500
Subject: [PATCH] Tweak loopy kernel stringification [ci skip]
---
grudge/symbolic/compiler.py | 10 ++++++++--
1 file changed, 8 insertions(+), 2 deletions(-)
diff --git a/grudge/symbolic/compiler.py b/grudge/symbolic/compiler.py
index 0d268ea4..cd82633a 100644
--- a/grudge/symbolic/compiler.py
+++ b/grudge/symbolic/compiler.py
@@ -108,7 +108,12 @@ class LoopyKernelInstruction(Instruction):
if isinstance(v, (Variable, Subscript)))
def __str__(self):
- knl_str = self.kernel_descriptor.loopy_kernel.stringify("di")
+ knl_str = "\n".join(
+ "%s = %s" % (insn.assignee, insn.expression)
+ for insn in self.kernel_descriptor.loopy_kernel.instructions)
+
+ knl_str = knl_str.replace("grdg_", "")
+
return "{ /* loopy */\n %s\n}" % knl_str.replace("\n", "\n ")
mapper_method = "map_insn_loopy_kernel"
@@ -277,7 +282,8 @@ def dot_dataflow_graph(code, max_node_label_length=30,
node_names[insn] = node_name
node_label = str(insn)
- if max_node_label_length is not None:
+ if (max_node_label_length is not None
+ and not isinstance(insn, LoopyKernelInstruction)):
node_label = node_label[:max_node_label_length]
if label_wrap_width is not None:
--
GitLab