@@ -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