blob: 2b3d0d63dd7b10e5179ac4567a81a08d4ed32b26 [file] [log] [blame]
# Copyright 2014 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import logging, threading, time
from autotest_lib.client.bin import utils
from autotest_lib.client.cros import service_stopper
class PerfControl(object):
Provides methods for setting the performance mode of a device.
In particular it verifies the machine is idle and cold and tries to set
it into a consistent, high performance state during initialization.
Furthermore it monitors the state of the machine (in particular
temperature) and verifies that nothing bad happened along the way.
Example usage:
with PerfControl() as pc:
if not pc.verify_is_valid():
raise error.TestError(pc.get_error_reason())
# Do all performance testing.
if not pc.verify_is_valid():
raise error.TestError(pc.get_error_reason())
def __init__(self):
self._service_stopper = None
# Keep a copy of the current state for cleanup.
self._temperature_init = utils.get_current_temperature_max()
self._throttle_count = 0
self._original_governors = utils.set_high_performance_mode()
self._error_reason = None
if not utils.wait_for_idle_cpu(60.0, 0.1):
self._error_reason = 'Could not get idle CPU.'
if not utils.wait_for_cool_machine():
self._error_reason = 'Could not get cold machine.'
self._temperature_cold = utils.get_current_temperature_max()
self._temperature_max = self._temperature_cold
# Should be last just in case we had a runaway process.
def __enter__(self):
return self
def __exit__(self, _type, value, traceback):
# First thing restart thermal management.
def get_error_reason(self):
Returns an error reason string if we encountered problems to pass
on to harness/wmatrix.
return self._error_reason
def verify_is_valid(self):
For now we declare performance results as valid if
- we did not have an error before.
- the monitoring thread never saw temperatures a throttle
TODO(ihf): Search log files for thermal throttling messages like in
if self._error_reason:
return False"Max observed temperature = %.1f'C (throttle_count=%d)",
self._temperature_max, self._throttle_count)
if (self._throttle_count):
self._error_reason = 'Machine got hot during testing.'
return False
return True
def _monitor_performance_state(self):
Checks machine temperature once per second.
TODO(ihf): make this more intelligent with regards to governor,
CPU, GPU and maybe zram as needed.
while True:
current_temperature = utils.get_current_temperature_max()
self._temperature_max = max(self._temperature_max,
is_throttled = utils.is_system_thermally_throttled()
if is_throttled:
self._throttle_count += 1
# TODO(ihf): Remove this spew once PerfControl is stable.'PerfControl system temperature = %.1f, throttled=%s',
current_temperature, is_throttled)
def _stop_thermal_throttling(self):
If exist on the platform/machine it stops the different thermal
throttling scripts from running.
Warning: this risks abnormal behavior if machine runs in high load.
self._service_stopper = service_stopper.get_thermal_service_stopper()
def _restore_thermal_throttling(self):
Restores the original thermal throttling state.
if self._service_stopper: