diff --git a/Run/measure_instance.py b/Run/measure_instance.py index bb9aa46d69c8da1f14067150786bad0dd485fb5d..3d278fb795040afd4c96c8bd06df5346e5bbd291 100644 --- a/Run/measure_instance.py +++ b/Run/measure_instance.py @@ -6,6 +6,7 @@ import json import time import expetator.experiment as experiment from expetator.monitors import Mojitos, kwollect +import platform from expetator.leverages import Dvfs, GpuClock #import run_flwr as run_flwr @@ -40,6 +41,22 @@ if instance_key not in config_data["instances"]: # Load config instance config = config_data["instances"][instance_key] +# Override the function i kwollect module +def get_g5k_target_metric(cluster_name=None): + if cluster_name is None: + cluster_name = platform.node().split('-')[0] + + if cluster_name in ['grisou', 'graoully', 'grimoire', + 'gros', 'gruss', 'paravance']: + return 'pdu_outlet_power_watt' + if cluster_name in ['troll', 'yeti', 'gemini', 'neowise', 'servan', 'sirius', 'paradoxe', + 'orion', 'pyxis', 'sagittaire', 'taurus', 'nova']: + return 'wattmetre_power_watt' + + return 'bmc_node_power_watt' + +kwollect.get_g5k_target_metric = get_g5k_target_metric + # extract DVFS infor (cpu and gpu) if "dvfs_cpu" in config: dvfs_config = config["dvfs_cpu"]