X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/dd05e77cf9fe231aef1608d42c8e45ac3ee5c15c..dc122c0fb3802061bc2fdb4abd50bf02a224cfdc:/src/smpi/internals/smpi_host.cpp

diff --git a/src/smpi/internals/smpi_host.cpp b/src/smpi/internals/smpi_host.cpp
index 95c7284f66..443c6859f8 100644
--- a/src/smpi/internals/smpi_host.cpp
+++ b/src/smpi/internals/smpi_host.cpp
@@ -1,10 +1,12 @@
-/* Copyright (c) 2017-2018. The SimGrid Team. All rights reserved.          */
+/* Copyright (c) 2017-2021. 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. */
 
 #include "smpi_host.hpp"
+#include "private.hpp"
 #include "simgrid/s4u/VirtualMachine.hpp"
+#include "smpi/smpi.h"
 #include "smpi_utils.hpp"
 #include "xbt/config.hpp"
 
@@ -14,13 +16,32 @@
 
 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(smpi_host, smpi, "Logging specific to SMPI (host)");
 
+/* save user's cost callbacks for SMPI operations */
+static std::unordered_map<SmpiOperation, SmpiOpCostCb> cost_cbs;
+
+void smpi_register_op_cost_callback(SmpiOperation op, const SmpiOpCostCb& cb)
+{
+  cost_cbs[op] = cb;
+}
+
+void smpi_cleanup_op_cost_callback()
+{
+  cost_cbs.clear();
+}
+
 namespace simgrid {
 namespace smpi {
 
-simgrid::xbt::Extension<simgrid::s4u::Host, Host> Host::EXTENSION_ID;
+xbt::Extension<s4u::Host, smpi::Host> Host::EXTENSION_ID;
 
-double Host::orecv(size_t size)
+double Host::orecv(size_t size, s4u::Host* src, s4u::Host* dst)
 {
+  /* return user's callback if available */
+  auto it = cost_cbs.find(SmpiOperation::RECV);
+  if (it != cost_cbs.end())
+    return it->second(size, src, dst);
+
+  /* fallback to smpi/or config */
   double current = orecv_parsed_values.empty() ? 0.0 : orecv_parsed_values.front().values[0] +
                                                            orecv_parsed_values.front().values[1] * size;
 
@@ -42,13 +63,18 @@ double Host::orecv(size_t size)
   return current;
 }
 
-double Host::osend(size_t size)
+double Host::osend(size_t size, s4u::Host* src, s4u::Host* dst)
 {
+  /* return user's callback if available */
+  auto it = cost_cbs.find(SmpiOperation::SEND);
+  if (it != cost_cbs.end())
+    return it->second(size, src, dst);
+
+  /* fallback to smpi/os config */
   double current =
       osend_parsed_values.empty() ? 0.0 : osend_parsed_values[0].values[0] + osend_parsed_values[0].values[1] * size;
-  // Iterate over all the sections that were specified and find the right
-  // value. (fact.factor represents the interval sizes; we want to find the
-  // section that has fact.factor <= size and no other such fact.factor <= size)
+  // Iterate over all the sections that were specified and find the right value. (fact.factor represents the interval
+  // sizes; we want to find the section that has fact.factor <= size and no other such fact.factor <= size)
   // Note: parse_factor() (used before) already sorts the vector we iterate over!
   for (auto const& fact : osend_parsed_values) {
     if (size <= fact.factor) { // Values already too large, use the previously computed value of current!
@@ -65,8 +91,14 @@ double Host::osend(size_t size)
   return current;
 }
 
-double Host::oisend(size_t size)
+double Host::oisend(size_t size, s4u::Host* src, s4u::Host* dst)
 {
+  /* return user's callback if available */
+  auto it = cost_cbs.find(SmpiOperation::ISEND);
+  if (it != cost_cbs.end())
+    return it->second(size, src, dst);
+
+  /* fallback to smpi/ois config */
   double current =
       oisend_parsed_values.empty() ? 0.0 : oisend_parsed_values[0].values[0] + oisend_parsed_values[0].values[1] * size;
 
@@ -88,33 +120,46 @@ double Host::oisend(size_t size)
   return current;
 }
 
-Host::Host(simgrid::s4u::Host *ptr) : host(ptr)
+void Host::check_factor_configs(const std::string& op) const
 {
-  if (not Host::EXTENSION_ID.valid())
-    Host::EXTENSION_ID = simgrid::s4u::Host::extension_create<Host>();
+  static const std::unordered_map<std::string, SmpiOperation> name_to_op_enum{
+      {"smpi/or", SmpiOperation::RECV}, {"smpi/os", SmpiOperation::SEND}, {"smpi/ois", SmpiOperation::ISEND}};
+  if (cost_cbs.find(name_to_op_enum.at(op)) != cost_cbs.end() &&
+      (host->get_property(op) || not config::is_default(op.c_str()))) {
+    XBT_WARN("SMPI (host: %s): mismatch cost functions for %s. Only user's callback will be used.", host->get_cname(),
+             op.c_str());
+  }
+}
 
+Host::Host(s4u::Host* ptr) : host(ptr)
+{
+  if (not smpi::Host::EXTENSION_ID.valid())
+    smpi::Host::EXTENSION_ID = s4u::Host::extension_create<Host>();
+
+  check_factor_configs("smpi/or");
   const char* orecv_string = host->get_property("smpi/or");
   if (orecv_string != nullptr) {
-    orecv_parsed_values = parse_factor(orecv_string);
+    orecv_parsed_values = simgrid::smpi::utils::parse_factor(orecv_string);
   } else {
-    orecv_parsed_values = parse_factor(simgrid::config::get_value<std::string>("smpi/or"));
+    orecv_parsed_values = simgrid::smpi::utils::parse_factor(config::get_value<std::string>("smpi/or"));
   }
 
+  check_factor_configs("smpi/os");
   const char* osend_string = host->get_property("smpi/os");
   if (osend_string != nullptr) {
-    osend_parsed_values = parse_factor(osend_string);
+    osend_parsed_values = simgrid::smpi::utils::parse_factor(osend_string);
   } else {
-    osend_parsed_values = parse_factor(simgrid::config::get_value<std::string>("smpi/os"));
+    osend_parsed_values = simgrid::smpi::utils::parse_factor(config::get_value<std::string>("smpi/os"));
   }
 
+  check_factor_configs("smpi/ois");
   const char* oisend_string = host->get_property("smpi/ois");
   if (oisend_string != nullptr) {
-    oisend_parsed_values = parse_factor(oisend_string);
+    oisend_parsed_values = simgrid::smpi::utils::parse_factor(oisend_string);
   } else {
-    oisend_parsed_values = parse_factor(simgrid::config::get_value<std::string>("smpi/ois"));
+    oisend_parsed_values = simgrid::smpi::utils::parse_factor(config::get_value<std::string>("smpi/ois"));
   }
 }
 
-Host::~Host()=default;
-}
-}
+} // namespace smpi
+} // namespace simgrid