commit 775e396e2939b55f5f1ea5261260533cf168d12c
author Michael McGeagh <michael.mcgeagh@arm.com> Tue Jul 28 11:44:22 2020 +0100
committer tim.hall <tim.hall@arm.com> Tue Aug 04 11:19:56 2020 +0000
tree fe3e42cbdb25443f11010e8278c704317bf3d730
parent 2da7d1116974de2f660c421314a4b59da6da87ee

test: Add extra print for verbose subgraphs

When using the various verbose options to print extra info, there is no
break in the output produced by vela.
Added the name of the function as part of the printing.
Added the name of the subgraph to distinguish between them.

Signed-off-by: Michael McGeagh <michael.mcgeagh@arm.com>
Change-Id: Ib489cf5043bd9d49b22c976afc545ee600965737

ethosu/vela/nn_graph.py

diff --git a/ethosu/vela/nn_graph.py b/ethosu/vela/nn_graph.py
index 247e6cc..1024307 100644
--- a/ethosu/vela/nn_graph.py
+++ b/ethosu/vela/nn_graph.py
@@ -303,9 +303,9 @@
         return all_ops
 
     def print_operators(self):
+        print("print_operators()", self.name)
         all_ops = self.get_all_ops()
         unique_ops = []
-        print("print_operators")
         for op in all_ops:
             if op.type in set(("Const", "Identity", "Placeholder")):
                 continue
@@ -341,11 +341,13 @@
                 print(s + " " + str(a))
 
     def print_graph(self):
+        print("print_graph()", self.name)
         all_ops = self.get_all_ops()
         for idx, op in enumerate(all_ops):
             print(idx, op.type, op.name)
 
     def print_graph_with_tensors(self):
+        print("print_graph_with_tensors()", self.name)
         all_ops = self.get_all_ops()
         for idx, op in enumerate(all_ops):
             print(idx, op.type, op.name)
@@ -362,6 +364,7 @@
             print()
 
     def print_graph_with_tensor_quantization(self):
+        print("print_graph_with_tensor_quantization()", self.name)
         all_ops = self.get_all_ops()
         for idx, op in enumerate(all_ops):
             print(idx, op.type, op.name)
@@ -386,10 +389,12 @@
             print()
 
     def print_passes(self):
+        print("print_passes()", self.name)
         for idx, ps in enumerate(self.passes):
             print("%03d %s" % (idx * 2, ps))
 
     def print_passes_with_tensors(self):
+        print("print_passes_with_tensors()", self.name)
         for idx, ps in enumerate(self.passes):
             print("%3d %s" % (idx * 2, ps))
             for idx, tens in enumerate(ps.inputs):
@@ -410,10 +415,12 @@
             print()
 
     def print_cascaded_passes(self):
+        print("print_cascaded_passes()", self.name)
         for idx, ps in enumerate(self.cascaded_passes):
             print("%3d %s SRAM used %.1f KB" % (idx * 2, ps, ps.sram_used / 1024))
 
     def print_cascaded_passes_with_tensors(self):
+        print("print_cascaded_passes_with_tensors()", self.name)
         for idx, ps in enumerate(self.cascaded_passes):
             print("%3d %s SRAM used %.1f KB" % (idx * 2, ps, ps.sram_used / 1024))
             for idx, tens in enumerate(ps.inputs):
@@ -434,6 +441,7 @@
             print()
 
     def print_cascaded_passes_with_tensor_sizes(self):
+        print("print_cascaded_passes_with_tensor_sizes()", self.name)
         for idx, ps in enumerate(self.cascaded_passes):
             print("%3d %s SRAM used %.1f KB" % (idx * 2, ps, ps.sram_used / 1024))
             for idx, tens in enumerate(ps.inputs):
@@ -478,6 +486,7 @@
             print()
 
     def print_high_level_command_stream(self):
+        print("print_high_level_command_stream()", self.name)
         for idx, cmd in enumerate(self.high_level_command_stream):
             print("%3d %s" % (idx, cmd))