runtime/onert/core/src/dumper/text/GraphDumper.cc

   1 /*
   2  * Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #include "GraphDumper.h"
  18
  19 #include "ir/Graph.h"
  20 #include "compiler/LoweredGraph.h"
  21 #include "util/logging.h"
  22 #include "misc/string_helpers.h"
  23
  24 namespace onert
  25 {
  26 namespace dumper
  27 {
  28 namespace text
  29 {
  30
  31 namespace
  32 {
  33
  34 std::string formatOperandIndexSequence(const ir::OperandIndexSequence &seq)
  35 {
  36   std::vector<std::string> strs;
  37   for (auto ind : seq)
  38     strs.push_back(dumper::text::formatOperandBrief(ind));
  39   return nnfw::misc::join(strs.begin(), strs.end(), ", ");
  40 }
  41
  42 } // namespace
  43
  44 std::string formatOperandBrief(ir::OperandIndex ind)
  45 {
  46   std::stringstream ss;
  47   ss << ind;
  48   return ss.str();
  49 }
  50
  51 std::string formatOperand(const ir::Graph &, ir::OperandIndex ind)
  52 {
  53   std::stringstream ss;
  54   ss << ind;
  55   // TODO Print shape, type and maybe more
  56   return ss.str();
  57 }
  58
  59 std::string formatOperation(const ir::Graph &graph, ir::OperationIndex ind)
  60 {
  61   std::stringstream ss;
  62   const auto &op = graph.operations().at(ind);
  63
  64   ss << formatOperandIndexSequence(op.getOutputs());
  65   ss << " = ";
  66   ss << ind << "_" << op.name() << "(";
  67   ss << formatOperandIndexSequence(op.getInputs());
  68   ss << ")";
  69   return ss.str();
  70 }
  71
  72 void dumpGraph(const ir::Graph &graph)
  73 {
  74   VERBOSE(GraphDumper) << "{\n";
  75   auto ops_topol = graph.topolSortOperations();
  76   for (auto op_ind : ops_topol)
  77   {
  78     VERBOSE(GraphDumper) << "  " << formatOperation(graph, op_ind) << "\n";
  79   }
  80   VERBOSE(GraphDumper) << "}\n";
  81   VERBOSE(GraphDumper) << std::endl;
  82 }
  83
  84 void dumpLoweredGraph(const compiler::LoweredGraph &lgraph)
  85 {
  86   // TODO Graph dump with backend info
  87   dumpGraph(lgraph.graph());
  88 }
  89
  90 } // namespace text
  91 } // namespace dumper
  92 } // namespace onert