-/* Copyright (c) 2014. The SimGrid Team.
+/* Copyright (c) 2014-2020. The SimGrid Team.
* All rights reserved. */
/* This program is free software; you can redistribute it and/or modify it
#ifndef SURF_NETWORK_IB_HPP_
#define SURF_NETWORK_IB_HPP_
-class NetworkIBModel : public NetworkModel {
-private:
-public:
- NetworkIBModel();
- NetworkIBModel(const char *name);
- ~NetworkModel();
- virtual ActionPtr communicate(RoutingEdgePtr src, RoutingEdgePtr dst,
- double size, double rate);
- virtual NetworkLinkPtr createNetworkLink(const char *name,
- double bw_initial,
- tmgr_trace_t bw_trace,
- double lat_initial,
- tmgr_trace_t lat_trace,
- e_surf_resource_state_t state_initial,
- tmgr_trace_t state_trace,
- e_surf_link_sharing_policy_t policy,
- xbt_dict_t properties);
+#include "src/surf/network_smpi.hpp"
+#include "xbt/base.h"
+
+#include <map>
+#include <vector>
+
+namespace simgrid {
+namespace kernel {
+namespace resource {
+
+class XBT_PRIVATE IBNode;
+
+struct XBT_PRIVATE ActiveComm {
+ IBNode* destination = nullptr;
+ NetworkAction* action = nullptr;
+ double init_rate = -1;
};
-class NetworkIBLink : public NetworkLink {
-private:
+class IBNode {
public:
- NetworkIBLink(NetworkModelPtr model, const char *name, xbt_dict_t props);
- NetworkIBLink(NetworkModelPtr model, const char *name, xbt_dict_t props,
- lmm_constraint_t constraint,
- tmgr_history_t history,
- tmgr_trace_t state_trace);
- ~NetworkIBLink();
- virtual void updateLatency(double value, double date=surf_get_clock());
- virtual void updateBandwidth(double value, double date=surf_get_clock());
+ int id;
+ // store related links, to ease computation of the penalties
+ std::vector<ActiveComm*> ActiveCommsUp;
+ // store the number of comms received from each node
+ std::map<IBNode*, int> ActiveCommsDown;
+ // number of comms the node is receiving
+ int nbActiveCommsDown = 0;
+ explicit IBNode(int id) : id(id){};
+ virtual ~IBNode() = default;
+};
+
+class XBT_PRIVATE NetworkIBModel : public NetworkSmpiModel {
+ double Bs;
+ double Be;
+ double ys;
+ void updateIBfactors_rec(IBNode* root, std::vector<bool>& updatedlist);
+ void computeIBfactors(IBNode* root);
+public:
+ NetworkIBModel();
+ explicit NetworkIBModel(const char* name);
+ NetworkIBModel(const NetworkIBModel&) = delete;
+ NetworkIBModel& operator=(const NetworkIBModel&) = delete;
+ void updateIBfactors(NetworkAction* action, IBNode* from, IBNode* to, int remove);
+ std::unordered_map<std::string, IBNode> active_nodes;
+ std::unordered_map<NetworkAction*, std::pair<IBNode*, IBNode*>> active_comms;
};
+} // namespace resource
+} // namespace kernel
+} // namespace simgrid
#endif