[locop] Show Push/Pull index field (#6132)

CanonicalNodeDescBuilder now shows the index attribute of Push/Pull
nodes.

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>

diff --git a/compiler/locop/src/FormattedGraph.cpp b/compiler/locop/src/FormattedGraph.cpp
index b2d78f4..a855e40 100644 (file)
--- a/compiler/locop/src/FormattedGraph.cpp
+++ b/compiler/locop/src/FormattedGraph.cpp
@@ -238,6 +238,7 @@ public:
   {
     NodeDesc res{opname(node)};
 
+    res.arg("index", node->indexed() ? pp::fmt(node->index()) : pp::fmt('?'));
     res.arg("from", symbol_lookup(*_symtbl, node->from()));
     res.arg("shape", pp::fmt(pretty(tensor_shape(node))));
     res.state(NodeDesc::State::Complete);
@@ -249,6 +250,7 @@ public:
   {
     NodeDesc res{opname(node)};
 
+    res.arg("index", node->indexed() ? pp::fmt(node->index()) : pp::fmt('?'));
     // TODO Print dtype
     res.arg("shape", pp::fmt(pretty(tensor_shape(node))));
     res.state(NodeDesc::State::PartiallyKnown);