examples/platforms/supernode.cpp

   1 /* Copyright (c) 2006-2023. The SimGrid Team. All rights reserved.          */
   2
   3 /* This program is free software; you can redistribute it and/or modify it
   4  * under the terms of the license (GNU LGPL) which comes with this package. */
   5
   6 #include <simgrid/s4u.hpp>
   7 namespace sg4 = simgrid::s4u;
   8
   9 constexpr double BW_CPU  = 1e12;
  10 constexpr double LAT_CPU = 0;
  11
  12 constexpr double BW_NODE  = 1e11;
  13 constexpr double LAT_NODE = 1e-8;
  14
  15 constexpr double BW_NETWORK  = 1.25e10;
  16 constexpr double LAT_NETWORK = 1e-7;
  17
  18 static std::string int_string(int n)
  19 {
  20   return simgrid::xbt::string_printf("%02d", n);
  21 }
  22
  23 /**
  24  *
  25  * This function creates one node made of nb_cpu CPUs.
  26  */
  27 static sg4::NetZone* create_node(const sg4::NetZone* root, const std::string& node_name, const int nb_cpu)
  28 {
  29   auto* node = sg4::create_star_zone(node_name)->set_parent(root);
  30
  31   /* create all hosts and connect them to outside world */
  32   for (int i = 0; i < nb_cpu; i++) {
  33     const auto& cpuname  = node_name + "_cpu-" + int_string(i);
  34     const auto& linkname = "link_" + cpuname;
  35
  36     const sg4::Host* host = node->create_host(cpuname, 1e9);
  37     const sg4::Link* l    = node->create_split_duplex_link(linkname, BW_CPU)->set_latency(LAT_CPU);
  38     node->add_route(host, nullptr, {{l, sg4::LinkInRoute::Direction::UP}}, true);
  39   }
  40
  41   return node;
  42 }
  43
  44 /**
  45  *
  46  * This function creates one super-node made of nb_nodes nodes with nb_cpu CPUs.
  47  */
  48 static sg4::NetZone* create_supernode(const sg4::NetZone* root, const std::string& supernode_name, const int nb_nodes,
  49                                       const int nb_cpu)
  50 {
  51   auto* supernode = sg4::create_star_zone(supernode_name)->set_parent(root);
  52
  53   /* create all nodes and connect them to outside world */
  54   for (int i = 0; i < nb_nodes; i++) {
  55     const auto& node_name = supernode_name + "_node-" + int_string(i);
  56     const auto& linkname  = "link_" + node_name;
  57
  58     sg4::NetZone* node = create_node(supernode, node_name, nb_cpu);
  59     node->set_gateway(node->create_router("router_" + node_name));
  60     node->seal();
  61
  62     const sg4::Link* l = supernode->create_split_duplex_link(linkname, BW_NODE)->set_latency(LAT_NODE);
  63     supernode->add_route(node, nullptr, {{l, sg4::LinkInRoute::Direction::UP}}, true);
  64   }
  65   return supernode;
  66 }
  67
  68 /**
  69  *
  70  * This function creates one cluster of nb_supernodes super-nodes made of nb_nodes nodes with nb_cpu CPUs.
  71  */
  72 static sg4::NetZone* create_cluster(const std::string& cluster_name, const int nb_supernodes, const int nb_nodes,
  73                                     const int nb_cpu)
  74 {
  75   auto* cluster = sg4::create_star_zone(cluster_name);
  76
  77   /* create all supernodes and connect them to outside world */
  78   for (int i = 0; i < nb_supernodes; i++) {
  79     const auto& supernode_name = cluster_name + "_supernode-" + int_string(i);
  80     const auto& linkname       = "link_" + supernode_name;
  81
  82     sg4::NetZone* supernode = create_supernode(cluster, supernode_name, nb_nodes, nb_cpu);
  83     supernode->set_gateway(supernode->create_router("router_" + supernode_name));
  84     supernode->seal();
  85
  86     const sg4::Link* l = cluster->create_split_duplex_link(linkname, BW_NETWORK)->set_latency(LAT_NETWORK);
  87     cluster->add_route(supernode, nullptr, {{l, sg4::LinkInRoute::Direction::UP}}, true);
  88   }
  89   return cluster;
  90 }
  91
  92 extern "C" void load_platform(const sg4::Engine& e);
  93 void load_platform(const sg4::Engine&)
  94 {
  95   create_cluster("cluster", 4, 6, 2)->seal();
  96 }