examples/platforms/supernode.cpp

   1 /* Copyright (c) 2006-2022. 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     return simgrid::xbt::string_printf("%02d", n);
  20 }
  21
  22 /**
  23  *
  24  * This function creates one node made of nb_cpu CPUs.
  25  */
  26 static sg4::NetZone *create_node(const sg4::NetZone* root, const std::string& node_name,
  27                                                      const int nb_cpu) {
  28     auto* node = sg4::create_star_zone(node_name);
  29     node->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)->seal();
  38
  39         node->add_route(host->get_netpoint(), nullptr, nullptr, nullptr, {{l, sg4::LinkInRoute::Direction::UP}}, true);
  40     }
  41
  42     return node;
  43 }
  44
  45 /**
  46  *
  47  * This function creates one super-node made of nb_nodes nodes with nb_cpu CPUs.
  48  */
  49 static sg4::NetZone *create_supernode(const sg4::NetZone* root, const std::string& supernode_name,
  50                                                      const int nb_nodes, const int nb_cpu) {
  51     auto* supernode = sg4::create_star_zone(supernode_name);
  52     supernode->set_parent(root);
  53
  54     /* create all nodes and connect them to outside world */
  55     for (int i=0; i<nb_nodes; i++) {
  56         const auto& node_name = supernode_name + "_node-" + int_string(i);
  57         const auto& linkname = "link_" + node_name;
  58
  59         sg4::NetZone *node = create_node(supernode, node_name, nb_cpu);
  60         const auto router = node->create_router("router_" + node_name);
  61         node->seal();
  62
  63         const sg4::Link* l = supernode->create_split_duplex_link(linkname, BW_NODE)->set_latency(LAT_NODE)->seal();
  64         supernode->add_route(node->get_netpoint(), nullptr, router, nullptr, {{l, sg4::LinkInRoute::Direction::UP}}, true);
  65     }
  66     return supernode;
  67 }
  68
  69 /**
  70  *
  71  * This function creates one cluster of nb_supernodes super-nodes made of nb_nodes nodes with nb_cpu CPUs.
  72  */
  73 static sg4::NetZone *create_cluster(const std::string& cluster_name, const int nb_supernodes,
  74                                                      const int nb_nodes, const int nb_cpu) {
  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         const auto router = 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)->seal();
  87         cluster->add_route(supernode->get_netpoint(), nullptr, router, 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& e)
  94 {
  95     create_cluster("cluster", 4, 6, 2)->seal();
  96 }