Add stream4 file

Author: jjotero

tutorials/basics/stream/stream4.py

@@ -0,0 +1,82 @@
+# Copyright 2016-2021 Swiss National Supercomputing Centre (CSCS/ETH Zurich)
+# ReFrame Project Developers. See the top-level LICENSE file for details.
+#
+# SPDX-License-Identifier: BSD-3-Clause
+
+import reframe as rfm
+import reframe.utility.sanity as sn
+
+
+@rfm.simple_test
+class StreamMultiSysTest(rfm.RegressionTest):
+    valid_systems = ['*']
+    valid_prog_environs = ['cray', 'gnu', 'intel', 'pgi']
+    prebuild_cmds = [
+        'wget http://www.cs.virginia.edu/stream/FTP/Code/stream.c',
+    ]
+    build_system = 'SingleSource'
+    sourcepath = 'stream.c'
+    variables = {
+        'OMP_NUM_THREADS': '4',
+        'OMP_PLACES': 'cores'
+    }
+    reference = {
+        'catalina': {
+            'Copy':  (25200, -0.05, 0.05, 'MB/s'),
+            'Scale': (16800, -0.05, 0.05, 'MB/s'),
+            'Add':   (18500, -0.05, 0.05, 'MB/s'),
+            'Triad': (18800, -0.05, 0.05, 'MB/s')
+        }
+    }
+
+    # Flags per programming environment
+    flags = variable(dict, value={
+        'cray':  ['-fopenmp', '-O3', '-Wall'],
+        'gnu':   ['-fopenmp', '-O3', '-Wall'],
+        'intel': ['-qopenmp', '-O3', '-Wall'],
+        'pgi':   ['-mp', '-O3']
+    })
+
+    # Number of cores for each system
+    cores = variable(dict, value={
+        'catalina:default': 4,
+        'daint:gpu': 12,
+        'daint:mc': 36,
+        'daint:login': 10
+    })
+
+    @run_before('compile')
+    def set_compiler_flags(self):
+        self.build_system.cppflags = ['-DSTREAM_ARRAY_SIZE=$((1 << 25))']
+        environ = self.current_environ.name
+        self.build_system.cflags = self.flags.get(environ, [])
+
+    @run_before('run')
+    def set_num_threads(self):
+        num_threads = self.cores.get(self.current_partition.fullname, 1)
+        self.num_cpus_per_task = num_threads
+        self.variables = {
+            'OMP_NUM_THREADS': str(num_threads),
+            'OMP_PLACES': 'cores'
+        }
+
+    @sanity_function
+    def validate_solution(self):
+        return sn.assert_found(r'Solution Validates', self.stdout)
+
+    @performance_function('MB/s')
+    def extract_bw(self, kind='Copy'):
+        if kind not in {'Copy', 'Scale', 'Add', 'Triad'}:
+            raise ValueError(f'illegal value in argument kind ({kind!r})')
+
+        return sn.extractsingle(rf'{kind}:\s+(\S+)\s+.*',
+                                self.stdout, 1, float)
+
+    @run_before('performance')
+    def set_perf_variables(self):
+        self.perf_variables = {
+            'Copy': self.extract_bw(),
+            'Scale': self.extract_bw('Scale'),
+            'Add': self.extract_bw('Add'),
+            'Triad': self.extract_bw('Triad'),
+        }