mirror of
https://github.com/ARM-software/workload-automation.git
synced 2025-10-22 03:34:08 +01:00
workloads: add memcpy workload
Ported memcpy workload from WA2.
This commit is contained in:
81
wa/workloads/memcpy/__init__.py
Normal file
81
wa/workloads/memcpy/__init__.py
Normal file
@@ -0,0 +1,81 @@
|
||||
# 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
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
#
|
||||
|
||||
# pylint: disable=E1101,W0201
|
||||
|
||||
import os
|
||||
import re
|
||||
|
||||
from wa import Workload, Parameter, Executable
|
||||
|
||||
|
||||
THIS_DIR = os.path.dirname(__file__)
|
||||
|
||||
|
||||
RESULT_REGEX = re.compile('Total time: ([\d.]+) s.*Bandwidth: ([\d.]+) MB/s', re.S)
|
||||
|
||||
|
||||
class Memcpy(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,
|
||||
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.
|
||||
'''),
|
||||
]
|
||||
|
||||
def initialize(self, context):
|
||||
self.binary_name = 'memcpy'
|
||||
resource = Executable(self, self.target.abi, self.binary_name)
|
||||
host_binary = context.resolver.get(resource)
|
||||
Memcpy.target_exe = self.target.install_if_needed(host_binary)
|
||||
|
||||
def setup(self, context):
|
||||
self.command = '{} -i {} -s {}'.format(self.target_exe, self.iterations, self.buffer_size)
|
||||
for c in (self.cpus or []):
|
||||
self.command += ' -c {}'.format(c)
|
||||
self.result = None
|
||||
|
||||
def run(self, context):
|
||||
self.result = self.target.execute(self.command, timeout=300)
|
||||
|
||||
def extract_results(self, context):
|
||||
if self.result:
|
||||
match = RESULT_REGEX.search(self.result)
|
||||
context.add_metric('time', float(match.group(1)), 'seconds', lower_is_better=True)
|
||||
context.add_metric('bandwidth', float(match.group(2)), 'MB/s')
|
Reference in New Issue
Block a user