# Copyright 2016-2020 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 Example5Test(rfm.RegressionTest):

def __init__(self):

self.descr = 'Matrix-vector multiplication example with CUDA'

self.valid_systems = ['daint:gpu']

self.valid_prog_environs = ['PrgEnv-cray', 'PrgEnv-gnu', 'PrgEnv-pgi']

self.sourcepath = 'example_matrix_vector_multiplication_cuda.cu'

self.executable_opts = ['1024', '100']

self.modules = ['cudatoolkit']

self.num_gpus_per_node = 1

self.sanity_patterns = sn.assert_found(

r'time for single matrix vector multiplication', self.stdout)

self.maintainers = ['you-can-type-your-email-here']

self.tags = {'tutorial'}