Source code for wlauto.workloads.memcpy

#    Copyright 2013-2015 ARM Limited
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# See the License for the specific language governing permissions and
# limitations under the License.

# pylint: disable=E1101,W0201

import os
import re

from wlauto import Workload, Parameter, Executable

THIS_DIR = os.path.dirname(__file__)

RESULT_REGEX = re.compile('Total time: ([\d.]+) s.*Bandwidth: ([\d.]+) MB/s', re.S)

[docs]class MemcpyTest(Workload): name = 'memcpy' description = """ Runs memcpy in a loop. This will run memcpy in a loop for a specified number of times on a buffer of a specified size. Additionally, the affinity of the test can be set to one or more specific cores. This workload is single-threaded. It genrates no scores or metrics by itself. """ parameters = [ Parameter('buffer_size', kind=int, default=1024 * 1024 * 5, description='Specifies the size, in bytes, of the buffer to be copied.'), Parameter('iterations', kind=int, default=1000, description='Specfies the number of iterations that will be performed.'), Parameter('cpus', kind=list, default=[], description="""A list of integers specifying ordinals of cores to which the affinity of the test process should be set. If not specified, all avaiable cores will be used. """), ]
[docs] def setup(self, context): self.binary_name = 'memcpy' host_binary = context.resolver.get(Executable(self, self.device.abi, self.binary_name)) self.device_binary = self.device.install_if_needed(host_binary) self.command = '{} -i {} -s {}'.format(self.device_binary, self.iterations, self.buffer_size) if self.cpus: for c in self.cpus: self.command += ' -c {}'.format(c)
[docs] def run(self, context): self.result = self.device.execute(self.command, timeout=300)
[docs] def update_result(self, context): match = context.result.add_metric('time', float(, 'seconds', lower_is_better=True) context.result.add_metric('bandwidth', float(, 'MB/s')
[docs] def teardown(self, context): pass