1 /* Copyright (c) 2014-2021. The SimGrid Team.
2 * All rights reserved. */
4 /* This program is free software; you can redistribute it and/or modify it
5 * under the terms of the license (GNU LGPL) which comes with this package. */
7 #ifndef SURF_NETWORK_IB_HPP_
8 #define SURF_NETWORK_IB_HPP_
10 #include "src/surf/network_smpi.hpp"
20 class XBT_PRIVATE IBNode;
22 struct XBT_PRIVATE ActiveComm {
23 IBNode* destination = nullptr;
24 NetworkAction* action = nullptr;
25 double init_rate = -1;
31 // store related links, to ease computation of the penalties
32 std::vector<ActiveComm*> active_comms_up_;
33 // store the number of comms received from each node
34 std::map<IBNode*, int> active_comms_down_;
35 // number of comms the node is receiving
36 int nb_active_comms_down_ = 0;
37 explicit IBNode(int id) : id_(id){};
40 class XBT_PRIVATE NetworkIBModel : public NetworkSmpiModel {
41 std::unordered_map<std::string, IBNode> active_nodes;
42 std::unordered_map<NetworkAction*, std::pair<IBNode*, IBNode*>> active_comms;
47 void update_IB_factors_rec(IBNode* root, std::vector<bool>& updatedlist) const;
48 void compute_IB_factors(IBNode* root) const;
51 explicit NetworkIBModel(const std::string& name);
52 NetworkIBModel(const NetworkIBModel&) = delete;
53 NetworkIBModel& operator=(const NetworkIBModel&) = delete;
54 void update_IB_factors(NetworkAction* action, IBNode* from, IBNode* to, int remove) const;
56 static void IB_create_host_callback(s4u::Host const& host);
57 static void IB_action_state_changed_callback(NetworkAction& action, Action::State /*previous*/);
58 static void IB_action_init_callback(NetworkAction& action);
60 } // namespace resource
62 } // namespace simgrid