sascha/devlib
1
0
Fork 0
mirror of https://github.com/ARM-software/devlib.git synced 2025-02-12 07:58:07 +00:00
Code Issues Releases Wiki Activity

module: gem5stats: add an iterator to match records

Browse Source
This commit is contained in:
Quentin Perret 2017-07-07 18:12:51 +01:00
parent f9bc6966c0
commit ce48ad217d
1 changed files with 28 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
devlib/module/gem5stats.py
View File

@ -28,6 +28,7 @@ class Gem5ROI:
self.target = target self.target = target
self.number = number self.number = number
self.running = False self.running = False
self.field = 'ROI::{}'.format(number)
def start(self): def start(self):
if self.running: if self.running:
@ -42,7 +43,7 @@ class Gem5ROI:
self.target.execute('m5 roiend {}'.format(self.number)) self.target.execute('m5 roiend {}'.format(self.number))
self.running = False self.running = False
return True return True
class Gem5StatsModule(Module): class Gem5StatsModule(Module):
''' '''
Module controlling Region of Interest (ROIs) markers, satistics dump Module controlling Region of Interest (ROIs) markers, satistics dump
@ -56,7 +57,7 @@ class Gem5StatsModule(Module):
@staticmethod @staticmethod
def probe(target): def probe(target):
return isinstance(target.platform, Gem5SimulationPlatform) return isinstance(target.platform, Gem5SimulationPlatform)
def __init__(self, target): def __init__(self, target):
super(Gem5StatsModule, self).__init__(target) super(Gem5StatsModule, self).__init__(target)
@ -109,6 +110,22 @@ class Gem5StatsModule(Module):
in-order list of records for the key under consideration during the in-order list of records for the key under consideration during the
active intervals of the ROI. active intervals of the ROI.
Keys must match fields in gem5's statistics log file. Key example:
system.cluster0.cores0.power_model.static_power
'''
records = defaultdict(lambda : defaultdict(list))
for record, active_rois in self.match_iter(keys, rois_labels):
for key in record:
for roi_label in active_rois:
records[key][roi_label].append(record[key])
return records
def match_iter(self, keys, rois_labels):
'''
Yields for each dump since origin a pair containing:
1. a dict storing the values corresponding to each of the specified keys
2. the list of currently active ROIs among those passed as parameters.
Keys must match fields in gem5's statistics log file. Key example: Keys must match fields in gem5's statistics log file. Key example:
system.cluster0.cores0.power_model.static_power system.cluster0.cores0.power_model.static_power
''' '''
@ -118,20 +135,19 @@ class Gem5StatsModule(Module):
if self.rois[label].running: if self.rois[label].running:
self.logger.warning('Trying to match records in statistics file' self.logger.warning('Trying to match records in statistics file'
' while ROI {} is running'.format(label)) ' while ROI {} is running'.format(label))
def roi_active(roi_label, dump):
roi = self.rois[roi_label]
return (roi.field in dump) and (int(dump[roi.field]) == 1)
records = {}
for key in keys:
records[key] = defaultdict(list)
with open(self._stats_file_path, 'r') as stats_file: with open(self._stats_file_path, 'r') as stats_file:
stats_file.seek(self._current_origin) stats_file.seek(self._current_origin)
for dump in iter_statistics_dump(stats_file): for dump in iter_statistics_dump(stats_file):
for label in rois_labels: active_rois = [l for l in rois_labels if roi_active(l, dump)]
# Save records only when ROIs are ON if active_rois:
roi_field = 'ROI::{}'.format(self.rois[label].number) record = {k: dump[k] for k in keys}
if (roi_field in dump) and (int(dump[roi_field]) == 1): yield (record, active_rois)
for key in keys:
records[key][label].append(dump[key])
return records
def reset_origin(self): def reset_origin(self):
''' '''