Method run

compiler/logo/src/Passes/RemoveDeadNodeWithQueryPass.cpp:29–70

{

bool RemoveDeadNodeWithQueryPass::run(loco::Graph *g)
{
  // Let's enumerate nodes required to compute output nodes
  auto active_nodes = loco::active_nodes(loco::output_nodes(g));

  // List dead(= non-active) nodes candidates
  std::set<loco::Node *> candidates;

  for (auto node : loco::all_nodes(g))
  {
    if (active_nodes.find(node) == active_nodes.end())
    {
      candidates.insert(node);
    }
  }

  // Find the nodes that should not be dead node in candidates
  for (auto it = candidates.begin(); it != candidates.end();)
  {
    if (auto service = (*it)->dialect()->service<DeadNodeQueryService>())
    {
      if (!service->isDeadNode(*it))
      {
        it = candidates.erase(it);
        continue;
      }
    }
    ++it;
  }

  for (auto node : candidates)
  {
    node->drop();
  }

  for (auto node : candidates)
  {
    g->nodes()->destroy(node);
  }

  return candidates.size() > 0;
}

} // namespace logo