sascha/workload-automation
1
0
Fork 0
mirror of https://github.com/ARM-software/workload-automation.git synced 2025-09-03 03:42:35 +01:00
Code Issues Releases Wiki Activity

Augmentations: Allow configuring of augmentations per workload

Browse Source 
Add support for enabling and disabling of augmentations (instruments and
output processors) on a per workload basis.
This commit is contained in:
Marc Bonnici
2018-01-29 09:47:52 +00:00
committed by setrofim
parent 557d62ce86
commit 9a556721b6
3 changed files with 99 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
wa/framework/job.py
View File

@@ -1,7 +1,7 @@
import logging
from datetime import datetime
from wa.framework import pluginloader, signal
from wa.framework import pluginloader, signal, instruments
from wa.framework.configuration.core import Status
# Because of use of Enum (dynamic attrs)
@@ -64,6 +64,33 @@ class Job(object):
self.set_status(Status.PENDING)
context.update_job_state(self)
def configure_augmentations(self, context, pm):
instruments_to_enable = set()
output_processors_to_enable = set()
enabled_instruments = set(i.name for i in instruments.get_enabled())
enabled_output_processors = set(p.name for p in pm.get_enabled())
for augmentation in self.spec.augmentations.values():
augmentation_cls = context.cm.plugin_cache.get_plugin_class(augmentation)
if augmentation_cls.kind == 'instrument':
instruments_to_enable.add(augmentation)
elif augmentation_cls.kind == 'output_processor':
output_processors_to_enable.add(augmentation)
# Disable unrequired instruments
for instrument in enabled_instruments.difference(instruments_to_enable):
instruments.disable(instrument)
# Enable additional instruments
for instrument in instruments_to_enable.difference(enabled_instruments):
instruments.enable(instrument)
# Disable unrequired output_processors
for processor in enabled_output_processors.difference(output_processors_to_enable):
pm.disable(processor)
# Enable additional output_processors
for processor in output_processors_to_enable.difference(enabled_output_processors):
pm.enable(processor)
def configure_target(self, context):
self.logger.info('Configuring target for job {} [{}]'.format(self.id, self.iteration))
context.tm.commit_runtime_parameters(self.spec.runtime_parameters)