Skip to content

Commit 7ebd4a5

Browse files
author
Vasileios Karakasis
committed
Fine tune GPU burn library test
1 parent 4265306 commit 7ebd4a5

File tree

3 files changed

+2
-3
lines changed

3 files changed

+2
-3
lines changed

config/cscs.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -533,7 +533,6 @@
533533
'descr': 'Tsa compute nodes',
534534
'max_jobs': 20,
535535
'features': ['gpu'],
536-
'extras': {'gpu_arch': 'sm_70'},
537536
'resources': [
538537
{
539538
'name': '_rfm_gpu',

hpctestlib/microbenchmarks/gpu/gpu_burn.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -185,7 +185,7 @@ def set_num_gpus_per_node(self):
185185

186186
gpu_devices = self.current_partition.select_devices('gpu')
187187
if gpu_devices:
188-
self.num_gpus_per_node = len(gpu_devices)
188+
self.num_gpus_per_node = gpu_devices[0].num_devices
189189

190190
@sanity_function
191191
def assert_sanity(self):

reframe/core/systems.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -400,7 +400,7 @@ def features(self):
400400
'''
401401
return self._features
402402

403-
def select_devices(devtype):
403+
def select_devices(self, devtype):
404404
'''Return all devices of the requested type:
405405
406406
:arg devtype: The type of the device info objects to return.

0 commit comments

Comments
 (0)