GPU burn nvidia draft

Author: Lewih

config_vsc.py

@@ -252,6 +252,10 @@
             'ftn': 'mpiifort',
             #'target_systems': ['vaughan', 'leibniz']
         },
+        {
+            'name': 'nvidia',
+            'modules': ['CUDA']
+        },
     ],
     'general': [
         {

tests/gpu/gpu_burn.py

@@ -0,0 +1,22 @@
+import os
+import reframe as rfm
+import reframe.utility.sanity as sn
+
+@rfm.simple_test
+class GPU_Burn_nvidia(rfm.RunOnlyRegressionTest):
+    descr = "GPU burn test on nvidia node"
+    valid_systems = ["vaughan:nvidia"]
+    valid_prog_environs = ["CUDA"]
+    variables = {'CUDAPATH': '/apps/antwerpen/rome/centos8/CUDA/11.6.2'}
+    time_limit = '20m'
+    num_tasks = 1
+    num_tasks_per_node = 1
+    num_cpus_per_task = 64
+    prebuild_cmds = ['git clone https://github.com/wilicc/gpu-burn.git']
+    prerun_cmds = ['cd gpu-burn', 'make']
+    executable = './gpu_burn 10'
+    tags = {"antwerp", "gpu", "gpuburn"}
+
+    @sanity_function
+    def assert_job(self):
+        return sn.and_(sn.assert_found(r'OK', self.stdout), sn.assert_not_found(r'FAULTY', self.stdout))