diff --git a/reframe_config_bot.py.tmpl b/reframe_config_bot.py.tmpl index 8f77429434..b63d5928b3 100644 --- a/reframe_config_bot.py.tmpl +++ b/reframe_config_bot.py.tmpl @@ -23,6 +23,10 @@ site_configuration = { FEATURES[GPU] ] + list(SCALES.keys()), 'resources': [ + { + 'name': '_rfm_gpu', + 'options': ['--gpus-per-node={num_gpus_per_node}'], + }, { 'name': 'memory', 'options': ['--mem={size}'], @@ -34,6 +38,12 @@ site_configuration = { 'mem_per_node': __MEM_PER_NODE__, GPU_VENDOR: GPU_VENDORS[NVIDIA], }, + 'devices': [ + { + 'type': DEVICE_TYPES[GPU], + 'num_devices': 4, + } + ], 'max_jobs': 1 } ]