Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
TorusZone: Add limiter for destination node
[simgrid.git] / src / kernel / routing / TorusZone.cpp
index 4aa65f7..c9c7299 100644 (file)
@@ -7,7 +7,6 @@
 #include "simgrid/kernel/routing/NetPoint.hpp"
 #include "simgrid/s4u/Host.hpp"
 #include "src/surf/network_interface.hpp"
-#include "src/surf/xml/platf_private.hpp"
 
 #include <boost/algorithm/string/classification.hpp>
 #include <boost/algorithm/string/split.hpp>
@@ -21,7 +20,7 @@ namespace simgrid {
 namespace kernel {
 namespace routing {
 
-void TorusZone::create_links_for_node(ClusterCreationArgs* cluster, int id, int rank, unsigned int position)
+void TorusZone::create_links(int id, int rank, unsigned int position)
 {
   /* Create all links that exist in the torus. Each rank creates @a dimensions-1 links */
   int dim_product = 1; // Needed to calculate the next neighbor_id
@@ -34,16 +33,19 @@ void TorusZone::create_links_for_node(ClusterCreationArgs* cluster, int id, int
                                ? rank - (current_dimension - 1) * dim_product
                                : rank + dim_product;
     // name of neighbor is not right for non contiguous cluster radicals (as id != rank in this case)
-    std::string link_id =
-        std::string(cluster->id) + "_link_from_" + std::to_string(id) + "_to_" + std::to_string(neighbor_rank_id);
+    std::string link_id = get_name() + "_link_from_" + std::to_string(id) + "_to_" + std::to_string(neighbor_rank_id);
     const s4u::Link* linkup;
     const s4u::Link* linkdown;
-    if (cluster->sharing_policy == s4u::Link::SharingPolicy::SPLITDUPLEX) {
-      linkup   = create_link(link_id + "_UP", std::vector<double>{cluster->bw})->set_latency(cluster->lat)->seal();
-      linkdown = create_link(link_id + "_DOWN", std::vector<double>{cluster->bw})->set_latency(cluster->lat)->seal();
+    if (get_link_sharing_policy() == s4u::Link::SharingPolicy::SPLITDUPLEX) {
+      linkup = create_link(link_id + "_UP", std::vector<double>{get_link_bandwidth()})
+                   ->set_latency(get_link_latency())
+                   ->seal();
+      linkdown = create_link(link_id + "_DOWN", std::vector<double>{get_link_bandwidth()})
+                     ->set_latency(get_link_latency())
+                     ->seal();
 
     } else {
-      linkup   = create_link(link_id, std::vector<double>{cluster->bw})->set_latency(cluster->lat)->seal();
+      linkup = create_link(link_id, std::vector<double>{get_link_bandwidth()})->set_latency(get_link_latency())->seal();
       linkdown = linkup;
     }
     /*
@@ -67,16 +69,12 @@ std::vector<unsigned int> TorusZone::parse_topo_parameters(const std::string& to
      * Parse attribute dimensions="dim1,dim2,dim3,...,dimN" and save them into a vector.
      * Additionally, we need to know how many ranks we have in total
      */
-    std::transform(begin(dimensions_str), end(dimensions_str), std::back_inserter(dimensions), surf_parse_get_int);
+    std::transform(begin(dimensions_str), end(dimensions_str), std::back_inserter(dimensions),
+                   [](const std::string& s) { return std::stoi(s); });
   }
   return dimensions;
 }
 
-void TorusZone::parse_specific_arguments(ClusterCreationArgs* cluster)
-{
-  set_topology(TorusZone::parse_topo_parameters(cluster->topo_parameters));
-}
-
 void TorusZone::set_topology(const std::vector<unsigned int>& dimensions)
 {
   xbt_assert(not dimensions.empty(), "Torus dimensions cannot be empty");
@@ -84,7 +82,7 @@ void TorusZone::set_topology(const std::vector<unsigned int>& dimensions)
   set_num_links_per_node(dimensions_.size());
 }
 
-void TorusZone::get_local_route(NetPoint* src, NetPoint* dst, RouteCreationArgs* route, double* lat)
+void TorusZone::get_local_route(NetPoint* src, NetPoint* dst, Route* route, double* lat)
 {
   XBT_VERB("torus getLocalRoute from '%s'[%u] to '%s'[%u]", src->get_cname(), src->id(), dst->get_cname(), dst->id());
 
@@ -94,7 +92,7 @@ void TorusZone::get_local_route(NetPoint* src, NetPoint* dst, RouteCreationArgs*
   if (src->id() == dst->id() && has_loopback()) {
     resource::LinkImpl* uplink = get_uplink_from(node_pos(src->id()));
 
-    route->link_list.push_back(uplink);
+    route->link_list_.push_back(uplink);
     if (lat)
       *lat += uplink->get_latency();
     return;
@@ -170,7 +168,7 @@ void TorusZone::get_local_route(NetPoint* src, NetPoint* dst, RouteCreationArgs*
     }
 
     if (has_limiter()) { // limiter for sender
-      route->link_list.push_back(get_uplink_from(node_pos_with_loopback(current_node)));
+      route->link_list_.push_back(get_uplink_from(node_pos_with_loopback(current_node)));
     }
 
     resource::LinkImpl* lnk;
@@ -179,75 +177,18 @@ void TorusZone::get_local_route(NetPoint* src, NetPoint* dst, RouteCreationArgs*
     else
       lnk = get_downlink_to(linkOffset);
 
-    route->link_list.push_back(lnk);
+    route->link_list_.push_back(lnk);
     if (lat)
       *lat += lnk->get_latency();
 
     current_node = next_node;
   }
-  // set gateways (if any)
-  route->gw_src = get_gateway(src->id());
-  route->gw_dst = get_gateway(dst->id());
-}
-
-/** @brief Auxiliary function to create hosts */
-static std::pair<kernel::routing::NetPoint*, kernel::routing::NetPoint*>
-create_torus_host(const kernel::routing::ClusterCreationArgs* cluster, s4u::NetZone* zone,
-                  const std::vector<unsigned int>& coord, int id)
-{
-  std::string host_id = std::string(cluster->prefix) + std::to_string(id) + cluster->suffix;
-  XBT_DEBUG("TorusCluster: creating host=%s speed=%f", host_id.c_str(), cluster->speeds.front());
-  s4u::Host* host = zone->create_host(host_id, cluster->speeds)
-                        ->set_core_count(cluster->core_amount)
-                        ->set_properties(cluster->properties)
-                        ->seal();
-  return std::make_pair(host->get_netpoint(), nullptr);
-}
-
-/** @brief Auxiliary function to create loopback links */
-static s4u::Link* create_torus_loopback(const kernel::routing::ClusterCreationArgs* cluster, s4u::NetZone* zone,
-                                        const std::vector<unsigned int>& coord, int id)
-{
-  std::string link_id = std::string(cluster->id) + "_link_" + std::to_string(id) + "_loopback";
-  XBT_DEBUG("TorusCluster: creating loopback link=%s bw=%f", link_id.c_str(), cluster->loopback_bw);
-
-  s4u::Link* loopback = zone->create_link(link_id, cluster->loopback_bw)
-                            ->set_sharing_policy(simgrid::s4u::Link::SharingPolicy::FATPIPE)
-                            ->set_latency(cluster->loopback_lat)
-                            ->seal();
-  return loopback;
-}
-
-/** @brief Auxiliary function to create limiter links */
-static s4u::Link* create_torus_limiter(const kernel::routing::ClusterCreationArgs* cluster, s4u::NetZone* zone,
-                                       const std::vector<unsigned int>& coord, int id)
-{
-  std::string link_id = std::string(cluster->id) + "_link_" + std::to_string(id) + "_limiter";
-  XBT_DEBUG("TorusCluster: creating limiter link=%s bw=%f", link_id.c_str(), cluster->limiter_link);
-
-  s4u::Link* limiter = zone->create_link(link_id, cluster->limiter_link)->seal();
-  return limiter;
-}
-
-s4u::NetZone* create_torus_zone_with_hosts(const kernel::routing::ClusterCreationArgs* cluster,
-                                           const s4u::NetZone* parent)
-{
-  using namespace std::placeholders;
-  auto set_host = std::bind(create_torus_host, cluster, _1, _2, _3);
-  std::function<s4u::TorusLinkCb> set_loopback{};
-  std::function<s4u::TorusLinkCb> set_limiter{};
-
-  if (cluster->loopback_bw > 0 || cluster->loopback_lat > 0) {
-    set_loopback = std::bind(create_torus_loopback, cluster, _1, _2, _3);
-  }
-
-  if (cluster->limiter_link > 0) {
-    set_loopback = std::bind(create_torus_limiter, cluster, _1, _2, _3);
+  if (has_limiter()) { // limiter for receiver/destination
+    route->link_list_.push_back(get_downlink_to(node_pos_with_loopback(dst->id())));
   }
-
-  return s4u::create_torus_zone(cluster->id, parent, TorusZone::parse_topo_parameters(cluster->topo_parameters),
-                                cluster->bw, cluster->lat, cluster->sharing_policy, set_host, set_loopback,
-                                set_limiter);
+  // set gateways (if any)
+  route->gw_src_ = get_gateway(src->id());
+  route->gw_dst_ = get_gateway(dst->id());
 }
 
 } // namespace routing
@@ -256,10 +197,8 @@ s4u::NetZone* create_torus_zone_with_hosts(const kernel::routing::ClusterCreatio
 namespace s4u {
 
 NetZone* create_torus_zone(const std::string& name, const NetZone* parent, const std::vector<unsigned int>& dimensions,
-                           double bandwidth, double latency, Link::SharingPolicy sharing_policy,
-                           const std::function<TorusNetPointCb>& set_netpoint,
-                           const std::function<TorusLinkCb>& set_loopback,
-                           const std::function<TorusLinkCb>& set_limiter)
+                           const ClusterCallbacks& set_callbacks, double bandwidth, double latency,
+                           Link::SharingPolicy sharing_policy)
 {
   int tot_elements = std::accumulate(dimensions.begin(), dimensions.end(), 1, std::multiplies<>());
   if (dimensions.empty() || tot_elements <= 0)
@@ -271,58 +210,20 @@ NetZone* create_torus_zone(const std::string& name, const NetZone* parent, const
     throw std::invalid_argument("TorusZone: incorrect latency for internode communication, lat=" +
                                 std::to_string(latency));
 
-  // auxiliary function to get dims from index
-  auto index_to_dims = [&dimensions](int index) {
-    std::vector<unsigned int> dims_array(dimensions.size());
-    for (int i = dimensions.size() - 1; i >= 0 && index > 0; --i) {
-      unsigned int value = index % dimensions[i];
-      dims_array[i]      = value;
-      index              = ((index / dimensions[i]));
-    }
-    return dims_array;
-  };
-
   auto* zone = new kernel::routing::TorusZone(name);
   zone->set_topology(dimensions);
   if (parent)
     zone->set_parent(parent->get_impl());
 
-  for (int i = 0; i < tot_elements; i++) {
-    kernel::routing::NetPoint *netpoint = nullptr, *gw = nullptr;
-    auto dims              = index_to_dims(i);
-    std::tie(netpoint, gw) = set_netpoint(zone->get_iface(), dims, i);
-    xbt_assert(netpoint, "TorusZone::set_netpoint(elem=%d): Invalid netpoint (nullptr)", i);
-    if (netpoint->is_netzone()) {
-      xbt_assert(gw && not gw->is_netzone(),
-                 "TorusZone::set_netpoint(elem=%d): Netpoint (%s) is a netzone, but gateway (%s) is invalid", i,
-                 netpoint->get_cname(), gw ? gw->get_cname() : "nullptr");
-    } else {
-      xbt_assert(not gw, "TorusZone: Netpoint (%s) isn't netzone, gateway must be nullptr", netpoint->get_cname());
-    }
-    // setting gateway
-    zone->set_gateway(i, gw);
-
-    if (set_loopback) {
-      Link* loopback = set_loopback(zone->get_iface(), dims, i);
-      xbt_assert(loopback, "TorusZone::set_loopback: Invalid loopback link (nullptr) for element %d", i);
-      zone->set_loopback();
-      zone->add_private_link_at(zone->node_pos(netpoint->id()), {loopback->get_impl(), loopback->get_impl()});
-    }
+  zone->set_link_characteristics(bandwidth, latency, sharing_policy);
 
-    if (set_limiter) {
-      Link* limiter = set_limiter(zone->get_iface(), dims, i);
-      xbt_assert(limiter, "TorusZone::set_limiter: Invalid limiter link (nullptr) for element %d", i);
-      zone->set_limiter();
-      zone->add_private_link_at(zone->node_pos_with_loopback(netpoint->id()),
-                                {limiter->get_impl(), limiter->get_impl()});
-    }
+  for (int i = 0; i < tot_elements; i++) {
+    kernel::routing::NetPoint* netpoint;
+    Link* limiter;
+    Link* loopback;
+    zone->fill_leaf_from_cb(i, dimensions, set_callbacks, &netpoint, &loopback, &limiter);
 
-    kernel::routing::ClusterCreationArgs params;
-    params.id             = name;
-    params.bw             = bandwidth;
-    params.lat            = latency;
-    params.sharing_policy = sharing_policy;
-    zone->create_links_for_node(&params, netpoint->id(), i, zone->node_pos_with_loopback_limiter(netpoint->id()));
+    zone->create_links(netpoint->id(), i, zone->node_pos_with_loopback_limiter(netpoint->id()));
   }
 
   return zone->get_iface();