From ec1fb83c4969aa78043b4b80417888b6cc1d0ee4 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Mon, 19 Dec 2016 16:12:39 +0100 Subject: [PATCH] Fix dependency graph drawing for {Call,ReturnFrom}Kernel --- loopy/kernel/tools.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/loopy/kernel/tools.py b/loopy/kernel/tools.py index 754507026..85a8da936 100644 --- a/loopy/kernel/tools.py +++ b/loopy/kernel/tools.py @@ -527,7 +527,9 @@ def get_dot_dependency_graph(kernel, iname_cluster=True, use_insn_id=False): lines.append("%s -> %s" % (insn_2, insn_1)) if iname_cluster: - from loopy.schedule import EnterLoop, LeaveLoop, RunInstruction, Barrier + from loopy.schedule import ( + EnterLoop, LeaveLoop, RunInstruction, Barrier, + CallKernel, ReturnFromKernel) for sched_item in kernel.schedule: if isinstance(sched_item, EnterLoop): @@ -537,7 +539,7 @@ def get_dot_dependency_graph(kernel, iname_cluster=True, use_insn_id=False): lines.append("}") elif isinstance(sched_item, RunInstruction): lines.append(sched_item.insn_id) - elif isinstance(sched_item, Barrier): + elif isinstance(sched_item, (CallKernel, ReturnFromKernel, Barrier)): pass else: raise LoopyError("schedule item not unterstood: %r" % sched_item) -- GitLab