Function print

src/utils/graph.cpp:37–48

namespace graph {

void print(const std::set<El::Int>& nodes,
           const std::map<El::Int, std::set<El::Int>>& edges,
           std::ostream& os = std::cout)
{
  for (const auto& node : nodes) {
    os << "node " << node << " neighbors :";
    for (const auto& neighbor : get_neighbors(node, edges)) {
      os << " " << neighbor;
    }
    os << "\n";
  }
}

std::set<El::Int>
get_neighbors(El::Int node, const std::map<El::Int, std::set<El::Int>>& edges)