opselector::OpSelector Class Reference

#include <OpSelector.h>

Public Member Functions
	OpSelector (const luci::Module *module)
template<SelectType SELECT_TYPE>
std::unique_ptr< luci::Module >	select_by (const std::string &str)

Detailed Description

Definition at line 31 of file OpSelector.h.

Constructor & Destructor Documentation

OpSelector()

opselector::OpSelector::OpSelector

(

const luci::Module *

module

)

Definition at line 265 of file OpSelector.cpp.

                                               : _module{module}
{
  // DO NOTHING
}

Member Function Documentation

select_by()

template<SelectType SELECT_TYPE>

template std::unique_ptr< luci::Module > opselector::OpSelector::select_by< SelectType::NAME >

(

const std::string &

str

)

Definition at line 376 of file OpSelector.cpp.

{
  auto colon_tokens = ::split_into_vector(str, ',');
  if (colon_tokens.empty())
  {
    throw std::runtime_error{"ERROR: Nothing was entered."};
  }
 
  auto selected_nodes = select_by<SELECT_TYPE>(colon_tokens);
 
  // multiout node should be considered
  IsMultiOutputNode multiout_visitor;
  std::vector<const luci::CircleNode *> output_nodes;
  for (const auto &node : selected_nodes)
  {
    if (node->accept(&multiout_visitor))
    {
      auto outputs = loco::succs(node);
      for (auto &o : outputs)
      {
        output_nodes.push_back(dynamic_cast<luci::CircleNode *>(o));
      }
    }
  }
  selected_nodes.insert(selected_nodes.end(), output_nodes.begin(), output_nodes.end());
 
  // TODO support two or more subgraphs
  for (const auto &n : selected_nodes)
  {
    IsMultiGraphNode multigraph_visitor;
    bool isMultiGraph = n->accept(&multigraph_visitor);
    if (isMultiGraph)
    {
      throw std::runtime_error{"ERROR: If or While operator can't be selected."};
    }
  }
 
  auto new_module = std::make_unique<luci::Module>();
  new_module->add(::make_graph(selected_nodes));
 
  return new_module;
}

References loco::succs().

---

The documentation for this class was generated from the following files:

• compiler/circle-opselector/src/OpSelector.h
• compiler/circle-opselector/src/OpSelector.cpp