Skip to content

Commit a6432eb

Browse files
committed
Make FFTW check generic
1 parent 0477b8b commit a6432eb

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

cscs-checks/microbenchmarks/fft/fftw_benchmark.py

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,13 +15,14 @@ def __init__(self, exec_mode):
1515
self.num_gpus_per_node = 0
1616
self.sanity_patterns = sn.assert_eq(
1717
sn.count(sn.findall(r'execution time', self.stdout)), 1)
18+
self.build_system.cflags = ['-O2']
1819
if self.current_system.name == 'kesch':
1920
self.valid_prog_environs = ['PrgEnv-cray', 'PrgEnv-pgi']
20-
self.build_system.cflags = ['-O2 -I$FFTW_INC -L$FFTW_DIR -lfftw3']
21-
else:
21+
self.build_system.cflags += ['-I$FFTW_INC', '-L$FFTW_DIR',
22+
'-lfftw3']
23+
elif self.current_system.name in {'daint', 'dom'}:
2224
self.valid_prog_environs = ['PrgEnv-cray', 'PrgEnv-pgi',
2325
'PrgEnv-gnu']
24-
self.build_system.cflags = ['-O2']
2526

2627
self.perf_patterns = {
2728
'fftw_exec_time': sn.extractsingle(
@@ -41,6 +42,9 @@ def __init__(self, exec_mode):
4142
},
4243
'kesch:cn': {
4344
'fftw_exec_time': (0.61, None, 0.05, 's'),
45+
},
46+
'*': {
47+
'fftw_exec_time': (0, None, None, 's'),
4448
}
4549
}
4650
else:
@@ -56,6 +60,9 @@ def __init__(self, exec_mode):
5660
'kesch:cn': {
5761
'fftw_exec_time': (1.58, None, 0.50, 's'),
5862
}
63+
'*': {
64+
'fftw_exec_time': (0, None, None, 's'),
65+
}
5966
}
6067

6168
self.maintainers = ['AJ']

0 commit comments

Comments
 (0)