diff --git a/benchpress/plugins/hooks/perf.py b/benchpress/plugins/hooks/perf.py index fc1243ce..78de15aa 100644 --- a/benchpress/plugins/hooks/perf.py +++ b/benchpress/plugins/hooks/perf.py @@ -11,6 +11,7 @@ import sys import traceback +from benchpress.lib import open_source from benchpress.lib.hook import Hook from .perf_monitors import ( @@ -25,6 +26,9 @@ vmstat, ) +if not open_source: + from .perf_monitors.fb_power import monitor as fb_power + BP_BASEPATH = os.path.dirname(os.path.abspath(sys.argv[0])) DEFAULT_OPTIONS = { @@ -45,6 +49,9 @@ "vmstat": {"interval": 5}, } +if not open_source: + DEFAULT_OPTIONS["fb_power"] = {"interval": 1} + AVAIL_MONITORS = { "mpstat": mpstat.MPStat, "cpufreq_scaling": cpufreq_scaling.CPUFreq, @@ -57,6 +64,9 @@ "vmstat": vmstat.VMStat, } +if not open_source: + AVAIL_MONITORS["fb_power"] = fb_power.FBPower + logger = logging.getLogger(__name__)