-/* Copyright (c) 2014-2018. The SimGrid Team. All rights reserved. */
+/* Copyright (c) 2014-2023. The SimGrid Team. All rights reserved. */
/* This program is free software; you can redistribute it and/or modify it
* under the terms of the license (GNU LGPL) which comes with this package. */
-#ifndef SURF_ROUTING_CLUSTER_DRAGONFLY_HPP_
-#define SURF_ROUTING_CLUSTER_DRAGONFLY_HPP_
+#ifndef SIMGRID_ROUTING_CLUSTER_DRAGONFLY_HPP_
+#define SIMGRID_ROUTING_CLUSTER_DRAGONFLY_HPP_
#include <simgrid/kernel/routing/ClusterZone.hpp>
+#include <simgrid/s4u/Link.hpp>
-namespace simgrid {
-namespace kernel {
-namespace routing {
+namespace simgrid::kernel::routing {
class DragonflyRouter {
public:
unsigned int group_;
unsigned int chassis_;
unsigned int blade_;
- surf::LinkImpl** blueLinks_ = nullptr;
- surf::LinkImpl** blackLinks_ = nullptr;
- surf::LinkImpl** greenLinks_ = nullptr;
- surf::LinkImpl** myNodes_ = nullptr;
- DragonflyRouter(int i, int j, int k);
- ~DragonflyRouter();
+ resource::StandardLinkImpl* blue_link_ = nullptr;
+ resource::StandardLinkImpl* limiter_ = nullptr;
+ std::vector<resource::StandardLinkImpl*> black_links_;
+ std::vector<resource::StandardLinkImpl*> green_links_;
+ std::vector<resource::StandardLinkImpl*> my_nodes_;
+ DragonflyRouter(unsigned group, unsigned chassis, unsigned blade, resource::StandardLinkImpl* limiter)
+ : group_(group), chassis_(chassis), blade_(blade), limiter_(limiter)
+ {
+ }
};
/** @ingroup ROUTING_API
* is also not realistic, as blue level can use more links than a single
* Aries can handle, thus it should use several routers.
*/
-class XBT_PUBLIC DragonflyZone : public ClusterZone {
+class XBT_PUBLIC DragonflyZone : public ClusterBase {
public:
- explicit DragonflyZone(NetZone* father, std::string name);
- ~DragonflyZone() override;
- // void create_links_for_node(sg_platf_cluster_cbarg_t cluster, int id, int rank, int position) override;
- void getLocalRoute(NetPoint* src, NetPoint* dst, RouteCreationArgs* into, double* latency) override;
- void parse_specific_arguments(ClusterCreationArgs* cluster) override;
- void seal() override;
- void generateRouters();
- void generateLinks();
- void createLink(const std::string& id, int numlinks, surf::LinkImpl** linkup, surf::LinkImpl** linkdown);
+ struct Coords {
+ unsigned long group;
+ unsigned long chassis;
+ unsigned long blade;
+ unsigned long node;
+ };
- void rankId_to_coords(int rankId, unsigned int (*coords)[4]);
+ explicit DragonflyZone(const std::string& name);
+ void get_local_route(const NetPoint* src, const NetPoint* dst, Route* into, double* latency) override;
+ /**
+ * @brief Parse topology parameters from string format
+ *
+ * @param topo_parameters Topology parameters, e.g. "3,4 ; 3,2 ; 3,1 ; 2"
+ */
+ static s4u::DragonflyParams parse_topo_parameters(const std::string& topo_parameters);
+
+ /** @brief Checks topology parameters */
+ static void check_topology(unsigned int n_groups, unsigned int groups_links, unsigned int n_chassis,
+ unsigned int chassis_links, unsigned int n_routers, unsigned int routers_links,
+ unsigned int nodes);
+ /** @brief Set Dragonfly topology */
+ void set_topology(unsigned int n_groups, unsigned int groups_links, unsigned int n_chassis,
+ unsigned int chassis_links, unsigned int n_routers, unsigned int routers_links, unsigned int nodes);
+ /** @brief Build upper levels (routers) in Dragonfly */
+ void build_upper_levels(const s4u::ClusterCallbacks& set_callbacks);
+ /** @brief Set the characteristics of links inside the Dragonfly zone */
+ void set_link_characteristics(double bw, double lat, s4u::Link::SharingPolicy sharing_policy) override;
+ Coords rankId_to_coords(unsigned long rank_id) const;
private:
- ClusterCreationArgs* cluster_ = nullptr;
- unsigned int numNodesPerBlade_ = 0;
- unsigned int numBladesPerChassis_ = 0;
- unsigned int numChassisPerGroup_ = 0;
- unsigned int numGroups_ = 0;
- unsigned int numLinksGreen_ = 0;
- unsigned int numLinksBlack_ = 0;
- unsigned int numLinksBlue_ = 0;
- unsigned int numLinksperLink_ = 1; // splitduplex -> 2, only for local link
- DragonflyRouter** routers_ = nullptr;
+ void generate_routers(const s4u::ClusterCallbacks& set_callbacks);
+ void generate_links();
+ void generate_link(const std::string& id, int numlinks, resource::StandardLinkImpl** linkup,
+ resource::StandardLinkImpl** linkdown);
+
+ unsigned int num_nodes_per_blade_ = 0;
+ unsigned int num_blades_per_chassis_ = 0;
+ unsigned int num_chassis_per_group_ = 0;
+ unsigned int num_groups_ = 0;
+ unsigned int num_links_green_ = 0;
+ unsigned int num_links_black_ = 0;
+ unsigned int num_links_blue_ = 0;
+ unsigned int num_links_per_link_ = 1; // splitduplex -> 2, only for local link
+ std::vector<DragonflyRouter> routers_;
};
-} // namespace routing
-} // namespace kernel
-} // namespace simgrid
+} // namespace simgrid::kernel::routing
#endif
