Skip to content

Commit c94acc9

Browse files
committed
test-perf Add and modify eval deriv perf tests
1 parent 05e89eb commit c94acc9

1 file changed

Lines changed: 8 additions & 5 deletions

File tree

tests/perf_tests.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -192,19 +192,22 @@ class EvalDerivs(TestParams):
192192
def __init__(self, ncores):
193193
# Overwrite certain ats options for GPU tests
194194
gpu_dict = dict(raja_test=True, np=1, nt=1, ngpu=1, nn=1)
195-
cpu_dict = dict(raja_test=True)
195+
raja_cpu_dict = dict(raja_test=True, np=int(ncores/8), nt=2)
196+
cpu_dict = dict(raja_test=False)
196197
super().__init__("EVALDERIV",
197198
"unit/SPH/evalDerivsRun.py",
198199
{"SOLIDSPH60": PerfTest(test_inps="--nx 60 --solid True", ats_inps=cpu_dict),
199-
"SOLIDSPHGPU60": PerfTest(test_inps="--nx 60 --solid True", ats_inps=gpu_dict),
200+
"SOLIDSPHGPU60": PerfTest(test_inps="--nx 60 --solid True --raja True", ats_inps=gpu_dict),
201+
"SOLIDSPHRAJA60": PerfTest(test_inps="--nx 60 --solid True --raja True", ats_inps=raja_cpu_dict),
200202
"SOLIDSPH80": PerfTest(test_inps="--nx 80 --solid True", ats_inps=cpu_dict),
201-
"SOLIDSPHGPU80": PerfTest(test_inps="--nx 80 --solid True", ats_inps=gpu_dict)})
203+
"SOLIDSPHGPU80": PerfTest(test_inps="--nx 80 --solid True --raja True", ats_inps=gpu_dict),
204+
"SOLIDSPHRAJA80": PerfTest(test_inps="--nx 80 --solid True --raja True", ats_inps=raja_cpu_dict)})
202205
# Only use half the number of cores
203-
self.ncores = int(ncores/2)
206+
self.ncores = int(ncores/4)
204207

205208
def set_gen_inputs(self):
206209
steps = 5
207-
self.gen_inp = f"--raja True --testDim 3d --steps {steps} --iterateH False --nPerh 2.01"
210+
self.gen_inp = f"--testDim 3d --steps {steps} --iterateH False --nPerh 2.01"
208211

209212
#---------------------------------------------------------------------------
210213
# General functions

0 commit comments

Comments
 (0)