/* 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 "energy.hpp"
-#include "../cpu_cas01.hpp"
-#include "../virtual_machine.hpp"
+#include <utility>
+#include <vector>
+
+#include <simgrid/plugins/energy.h>
+#include <simgrid/simix.hpp>
+#include <src/surf/plugins/energy.hpp>
+#include <src/surf/cpu_interface.hpp>
+#include <src/surf/virtual_machine.hpp>
/** @addtogroup SURF_plugin_energy
XBT_LOG_NEW_DEFAULT_SUBCATEGORY(surf_energy, surf,
"Logging specific to the SURF energy plugin");
-namespace simgrid {
-namespace energy {
-
-std::map<simgrid::surf::Host*, HostEnergy*> *surf_energy = NULL;
-
-}
-}
-
using simgrid::energy::HostEnergy;
-using simgrid::energy::surf_energy;
-static void energyHostCreatedCallback(simgrid::surf::Host *host){
- (*surf_energy)[host] = new HostEnergy(host);
-}
+namespace simgrid {
+namespace energy {
-static void energyVMCreatedCallback(simgrid::surf::VirtualMachine* vm) {
- std::map<simgrid::surf::Host*, HostEnergy*>::iterator host_energy_it = surf_energy->find(vm->p_subWs);
- xbt_assert(host_energy_it != surf_energy->end(), "The host is not in surf_energy.");
- (*surf_energy)[vm] = host_energy_it->second;
- host_energy_it->second->ref(); // protect the HostEnergy from getting deleted too early
-}
+simgrid::xbt::Extension<simgrid::s4u::Host, HostEnergy> HostEnergy::EXTENSION_ID;
/* Computes the consumption so far. Called lazily on need. */
-static void update_consumption(simgrid::surf::Host *host, HostEnergy *host_energy) {
- double cpu_load = lmm_constraint_get_usage(host->p_cpu->getConstraint()) / host->p_cpu->m_speedPeak;
- double start_time = host_energy->last_updated;
+void HostEnergy::update()
+{
+ simgrid::surf::Host* surf_host = host->extension<simgrid::surf::Host>();
+ double start_time = this->last_updated;
double finish_time = surf_get_clock();
+ double cpu_load;
+ if (surf_host->p_cpu->m_speedPeak == 0)
+ // Some users declare a pstate of speed 0 flops (eg to model boot time).
+ // We consider that the machine is then fully loaded. That's arbitrary but it avoids a NaN
+ cpu_load = 1;
+ else
+ cpu_load = lmm_constraint_get_usage(surf_host->p_cpu->getConstraint())
+ / surf_host->p_cpu->m_speedPeak;
- double previous_energy = host_energy->total_energy;
+ if (cpu_load > 1) // A machine with a load > 1 consumes as much as a fully loaded machine, not mores
+ cpu_load = 1;
+
+ double previous_energy = this->total_energy;
double instantaneous_consumption;
- if (host->getState() == SURF_RESOURCE_OFF)
- instantaneous_consumption = host_energy->watts_off;
+ if (host->is_off())
+ instantaneous_consumption = this->watts_off;
else
- instantaneous_consumption = host_energy->getCurrentWattsValue(cpu_load);
+ instantaneous_consumption = this->getCurrentWattsValue(cpu_load);
double energy_this_step = instantaneous_consumption*(finish_time-start_time);
- host_energy->total_energy = previous_energy + energy_this_step;
- host_energy->last_updated = finish_time;
+ this->total_energy = previous_energy + energy_this_step;
+ this->last_updated = finish_time;
- XBT_DEBUG("[cpu_update_energy] period=[%.2f-%.2f]; current power peak=%.0E flop/s; consumption change: %.2f J -> %.2f J",
- start_time, finish_time, host->p_cpu->m_speedPeak, previous_energy, energy_this_step);
+ XBT_DEBUG("[update_energy of %s] period=[%.2f-%.2f]; current power peak=%.0E flop/s; consumption change: %.2f J -> %.2f J",
+ surf_host->getName(), start_time, finish_time, surf_host->p_cpu->m_speedPeak, previous_energy, energy_this_step);
}
-static void energyHostDestructedCallback(simgrid::surf::Host *host){
- std::map<simgrid::surf::Host*, HostEnergy*>::iterator host_energy_it = surf_energy->find(host);
- xbt_assert(host_energy_it != surf_energy->end(), "The host is not in surf_energy.");
-
- HostEnergy *host_energy = host_energy_it->second;
- update_consumption(host, host_energy);
-
- if (host_energy_it->second->refcount == 1) // Don't display anything for virtual CPUs
- XBT_INFO("Total energy of host %s: %f Joules", host->getName(), host_energy->getConsumedEnergy());
- host_energy_it->second->unref();
- surf_energy->erase(host_energy_it);
-}
-
-static void energyCpuActionStateChangedCallback(
- simgrid::surf::CpuAction *action, e_surf_action_state_t old, e_surf_action_state_t cur)
+HostEnergy::HostEnergy(simgrid::s4u::Host *ptr) :
+ host(ptr), last_updated(surf_get_clock())
{
- const char *name = getActionCpu(action)->getName();
- simgrid::surf::Host *host = static_cast<simgrid::surf::Host*>(surf_host_resource_priv(sg_host_by_name(name)));
-
- HostEnergy *host_energy = (*surf_energy)[host];
+ initWattsRangeList();
+
+ if (host->properties() != NULL) {
+ char* off_power_str = (char*)xbt_dict_get_or_null(
+ host->properties(), "watt_off");
+ if (off_power_str != NULL)
+ watts_off = atof(off_power_str);
+ else
+ watts_off = 0;
+ }
- if(host_energy->last_updated < surf_get_clock())
- update_consumption(host, host_energy);
}
-static void energyStateChangedCallback(simgrid::surf::Host *host, e_surf_resource_state_t oldState, e_surf_resource_state_t newState)
+HostEnergy::~HostEnergy()
{
- HostEnergy *host_energy = (*surf_energy)[host];
-
- if(host_energy->last_updated < surf_get_clock())
- update_consumption(host, host_energy);
}
-static void sg_energy_plugin_exit()
+double HostEnergy::getWattMinAt(int pstate)
{
- delete surf_energy;
- surf_energy = NULL;
+ xbt_assert(!power_range_watts_list.empty(),
+ "No power range properties specified for host %s", host->name().c_str());
+ return power_range_watts_list[pstate].first;
}
-/** \ingroup SURF_plugin_energy
- * \brief Enable energy plugin
- * \details Enable energy plugin to get joules consumption of each cpu. You should call this function before #MSG_init().
- */
-void sg_energy_plugin_init() {
- if (simgrid::energy::surf_energy == NULL) {
- simgrid::energy::surf_energy =
- new std::map<simgrid::surf::Host*, simgrid::energy::HostEnergy*>();
- surf_callback_connect(simgrid::surf::hostCreatedCallbacks, energyHostCreatedCallback);
- surf_callback_connect(simgrid::surf::VMCreatedCallbacks, energyVMCreatedCallback);
- surf_callback_connect(simgrid::surf::hostDestructedCallbacks, energyHostDestructedCallback);
- surf_callback_connect(simgrid::surf::cpuActionStateChangedCallbacks, energyCpuActionStateChangedCallback);
- surf_callback_connect(simgrid::surf::surfExitCallbacks, sg_energy_plugin_exit);
- surf_callback_connect(simgrid::surf::hostStateChangedCallbacks, energyStateChangedCallback);
- }
-}
-
-namespace simgrid {
-namespace energy {
-
-/**
- *
- */
-HostEnergy::HostEnergy(simgrid::surf::Host *ptr)
+double HostEnergy::getWattMaxAt(int pstate)
{
- host = ptr;
- total_energy = 0;
- power_range_watts_list = getWattsRangeList();
- last_updated = surf_get_clock();
-
- if (host->getProperties() != NULL) {
- char* off_power_str = (char*)xbt_dict_get_or_null(host->getProperties(), "watt_off");
- if (off_power_str != NULL)
- watts_off = atof(off_power_str);
- else
- watts_off = 0;
- }
-
-}
-
-HostEnergy::~HostEnergy(){
- unsigned int iter;
- xbt_dynar_t power_tuple = NULL;
- xbt_dynar_foreach(power_range_watts_list, iter, power_tuple)
- xbt_dynar_free(&power_tuple);
- xbt_dynar_free(&power_range_watts_list);
-}
-
-
-double HostEnergy::getWattMinAt(int pstate) {
- xbt_dynar_t power_range_list = power_range_watts_list;
- xbt_assert(power_range_watts_list, "No power range properties specified for host %s", host->getName());
- xbt_dynar_t current_power_values = xbt_dynar_get_as(power_range_list, static_cast<simgrid::surf::CpuCas01*>(host->p_cpu)->getPState(), xbt_dynar_t);
- double min_power = xbt_dynar_get_as(current_power_values, 0, double);
- return min_power;
-}
-double HostEnergy::getWattMaxAt(int pstate) {
- xbt_dynar_t power_range_list = power_range_watts_list;
- xbt_assert(power_range_watts_list, "No power range properties specified for host %s", host->getName());
- xbt_dynar_t current_power_values = xbt_dynar_get_as(power_range_list, static_cast<simgrid::surf::CpuCas01*>(host->p_cpu)->getPState(), xbt_dynar_t);
- double max_power = xbt_dynar_get_as(current_power_values, 1, double);
- return max_power;
+ xbt_assert(!power_range_watts_list.empty(),
+ "No power range properties specified for host %s", host->name().c_str());
+ return power_range_watts_list[pstate].second;
}
-/**
- * Computes the power consumed by the host according to the current pstate and processor load
- *
- */
+/** @brief Computes the power consumed by the host according to the current pstate and processor load */
double HostEnergy::getCurrentWattsValue(double cpu_load)
{
- xbt_dynar_t power_range_list = power_range_watts_list;
- xbt_assert(power_range_watts_list, "No power range properties specified for host %s", host->getName());
-
- /* retrieve the power values associated with the current pstate */
- xbt_dynar_t current_power_values = xbt_dynar_get_as( power_range_list,
- static_cast<simgrid::surf::CpuCas01*>(host->p_cpu)->getPState(),
- xbt_dynar_t);
-
- /* min_power corresponds to the idle power (cpu load = 0) */
- /* max_power is the power consumed at 100% cpu load */
- double min_power = xbt_dynar_get_as(current_power_values, 0, double);
- double max_power = xbt_dynar_get_as(current_power_values, 1, double);
- double power_slope = max_power - min_power;
+ xbt_assert(!power_range_watts_list.empty(),
+ "No power range properties specified for host %s", host->name().c_str());
- double current_power = min_power + cpu_load * power_slope;
+ /* min_power corresponds to the idle power (cpu load = 0) */
+ /* max_power is the power consumed at 100% cpu load */
+ auto range = power_range_watts_list.at(host->pstate());
+ double min_power = range.first;
+ double max_power = range.second;
+ double power_slope = max_power - min_power;
+ double current_power = min_power + cpu_load * power_slope;
XBT_DEBUG("[get_current_watts] min_power=%f, max_power=%f, slope=%f", min_power, max_power, power_slope);
- XBT_DEBUG("[get_current_watts] Current power (watts) = %f, load = %f", current_power, cpu_load);
+ XBT_DEBUG("[get_current_watts] Current power (watts) = %f, load = %f", current_power, cpu_load);
return current_power;
}
double HostEnergy::getConsumedEnergy()
{
+ if (last_updated < surf_get_clock()) // We need to simcall this as it modifies the environment
+ simgrid::simix::kernel(std::bind(&HostEnergy::update, this));
- if(last_updated < surf_get_clock())
- update_consumption(host, this);
return total_energy;
-
}
-xbt_dynar_t HostEnergy::getWattsRangeList()
+void HostEnergy::initWattsRangeList()
{
- xbt_dynar_t power_range_list;
- xbt_dynar_t power_tuple;
- int i = 0, pstate_nb=0;
- xbt_dynar_t current_power_values;
- double min_power, max_power;
-
- if (host->getProperties() == NULL)
- return NULL;
-
- char* all_power_values_str = (char*)xbt_dict_get_or_null(host->getProperties(), "watt_per_state");
-
+ if (host->properties() == NULL)
+ return;
+ char* all_power_values_str =
+ (char*)xbt_dict_get_or_null(host->properties(), "watt_per_state");
if (all_power_values_str == NULL)
- return NULL;
-
+ return;
- power_range_list = xbt_dynar_new(sizeof(xbt_dynar_t), NULL);
xbt_dynar_t all_power_values = xbt_str_split(all_power_values_str, ",");
+ int pstate_nb = xbt_dynar_length(all_power_values);
- pstate_nb = xbt_dynar_length(all_power_values);
- for (i=0; i< pstate_nb; i++)
+ for (int i=0; i< pstate_nb; i++)
{
/* retrieve the power values associated with the current pstate */
- current_power_values = xbt_str_split(xbt_dynar_get_as(all_power_values, i, char*), ":");
+ xbt_dynar_t current_power_values = xbt_str_split(xbt_dynar_get_as(all_power_values, i, char*), ":");
xbt_assert(xbt_dynar_length(current_power_values) > 1,
- "Power properties incorrectly defined - could not retrieve min and max power values for host %s",
- host->getName());
+ "Power properties incorrectly defined - "
+ "could not retrieve min and max power values for host %s",
+ host->name().c_str());
/* min_power corresponds to the idle power (cpu load = 0) */
/* max_power is the power consumed at 100% cpu load */
- min_power = atof(xbt_dynar_get_as(current_power_values, 0, char*));
- max_power = atof(xbt_dynar_get_as(current_power_values, 1, char*));
-
- power_tuple = xbt_dynar_new(sizeof(double), NULL);
- xbt_dynar_push_as(power_tuple, double, min_power);
- xbt_dynar_push_as(power_tuple, double, max_power);
-
- xbt_dynar_push_as(power_range_list, xbt_dynar_t, power_tuple);
- xbt_dynar_free(¤t_power_values);
+ power_range_watts_list.push_back(power_range(
+ atof(xbt_dynar_get_as(current_power_values, 0, char*)),
+ atof(xbt_dynar_get_as(current_power_values, 1, char*))
+ ));
}
xbt_dynar_free(&all_power_values);
- return power_range_list;
}
}
}
+
+/* **************************** events callback *************************** */
+static void onCreation(simgrid::s4u::Host& host) {
+ simgrid::surf::Host* surf_host = host.extension<simgrid::surf::Host>();
+ if (dynamic_cast<simgrid::surf::VirtualMachine*>(surf_host)) // Ignore virtual machines
+ return;
+ host.extension_set(new HostEnergy(&host));
+}
+
+static void onActionStateChange(simgrid::surf::CpuAction *action, e_surf_action_state_t previous) {
+ const char *name = getActionCpu(action)->getName();
+ simgrid::surf::Host *host = sg_host_by_name(name)->extension<simgrid::surf::Host>();
+ simgrid::surf::VirtualMachine *vm = dynamic_cast<simgrid::surf::VirtualMachine*>(host);
+ if (vm) // If it's a VM, take the corresponding PM
+ host = vm->getPm()->extension<simgrid::surf::Host>();
+
+ HostEnergy *host_energy = host->p_host->extension<HostEnergy>();
+
+ if(host_energy->last_updated < surf_get_clock())
+ host_energy->update();
+}
+
+static void onHostStateChange(simgrid::s4u::Host &host) {
+ simgrid::surf::Host* surf_host = host.extension<simgrid::surf::Host>();
+ if (dynamic_cast<simgrid::surf::VirtualMachine*>(surf_host)) // Ignore virtual machines
+ return;
+
+ HostEnergy *host_energy = host.extension<HostEnergy>();
+
+ if(host_energy->last_updated < surf_get_clock())
+ host_energy->update();
+}
+
+static void onHostDestruction(simgrid::s4u::Host& host) {
+ // Ignore virtual machines
+ simgrid::surf::Host* surf_host = host.extension<simgrid::surf::Host>();
+ if (dynamic_cast<simgrid::surf::VirtualMachine*>(surf_host))
+ return;
+ HostEnergy *host_energy = host.extension<HostEnergy>();
+ host_energy->update();
+ XBT_INFO("Total energy of host %s: %f Joules",
+ host.name().c_str(), host_energy->getConsumedEnergy());
+}
+
+/* **************************** Public interface *************************** */
+/** \ingroup SURF_plugin_energy
+ * \brief Enable energy plugin
+ * \details Enable energy plugin to get joules consumption of each cpu. You should call this function before #MSG_init().
+ */
+void sg_energy_plugin_init(void)
+{
+ if (HostEnergy::EXTENSION_ID.valid())
+ return;
+
+ HostEnergy::EXTENSION_ID = simgrid::s4u::Host::extension_create<HostEnergy>();
+
+ simgrid::s4u::Host::onCreation.connect(&onCreation);
+ simgrid::s4u::Host::onStateChange.connect(&onHostStateChange);
+ simgrid::s4u::Host::onDestruction.connect(&onHostDestruction);
+ simgrid::surf::CpuAction::onStateChange.connect(&onActionStateChange);
+}
+
+/** @brief Returns the total energy consumed by the host so far (in Joules)
+ *
+ * See also @ref SURF_plugin_energy.
+ */
+double sg_host_get_consumed_energy(sg_host_t host) {
+ xbt_assert(HostEnergy::EXTENSION_ID.valid(),
+ "The Energy plugin is not active. "
+ "Please call sg_energy_plugin_init() during initialization.");
+ return host->extension<HostEnergy>()->getConsumedEnergy();
+}
+
+/** @brief Get the amount of watt dissipated at the given pstate when the host is idling */
+double sg_host_get_wattmin_at(sg_host_t host, int pstate) {
+ xbt_assert(HostEnergy::EXTENSION_ID.valid(),
+ "The Energy plugin is not active. "
+ "Please call sg_energy_plugin_init() during initialization.");
+ return host->extension<HostEnergy>()->getWattMinAt(pstate);
+}
+/** @brief Returns the amount of watt dissipated at the given pstate when the host burns CPU at 100% */
+double sg_host_get_wattmax_at(sg_host_t host, int pstate) {
+ xbt_assert(HostEnergy::EXTENSION_ID.valid(),
+ "The Energy plugin is not active. "
+ "Please call sg_energy_plugin_init() during initialization.");
+ return host->extension<HostEnergy>()->getWattMaxAt(pstate);
+}
