@@ -96,16 +96,21 @@ def prepare_build(self):
9696 )
9797 self .prebuild_cmds = [f'tar --strip-components=1 -xzf { tarsource } ' ]
9898 self .build_system .config_opts = [
99- '-DREGRESSIONTEST_DOWNLOAD=ON' ,
100- '-DGMX_MPI=on' ,
101- '-DGMX_BUILD_OWN_FFTW=ON' ,
102- '-DCP2K_USE_SPGLIB=ON' ,
99+ '-DREGRESSIONTEST_DOWNLOAD=OFF' ,
100+ '-DGMX_MPI=ON' ,
101+ '-DGMX_BUILD_OWN_FFTW=OFF' ,
103102 '-DGMX_HWLOC=ON' ,
104- '-DGMX_SIMD=ARM_NEON_ASIMD' ,
105103 '-DGMX_INSTALL_NBLIB_API=ON' ,
106104 ]
107105 if self .uarch == 'gh200' :
108- self .build_system .config_opts += ['-DGMX_GPU=CUDA' ]
106+ self .build_system .config_opts += [
107+ '-DGMX_SIMD=ARM_NEON_ASIMD' ,
108+ '-DGMX_GPU=CUDA'
109+ ]
110+ elif self .uarch == 'zen2' :
111+ self .build_system .config_opts += [
112+ '-DGMX_SIMD=AUTO' ,
113+ ]
109114
110115 @sanity_function
111116 def validate_test (self ):
@@ -115,7 +120,7 @@ def validate_test(self):
115120
116121@rfm .simple_test
117122class gromacs_run_test (rfm .RunOnlyRegressionTest ):
118- executable = './mps-wrapper.sh -- gmx_mpi mdrun -s stmv2.tpr'
123+ executable = 'gmx_mpi mdrun -s stmv2.tpr'
119124 executable_opts = [
120125 '-dlb no' , '-npme 1' , '-pin off' , '-v' , '-noconfout' , '-nstlist 300'
121126 ]
@@ -143,6 +148,8 @@ def prepare_run(self):
143148 self .executable_opts .append (f'-ntomp { self .num_cpus_per_task } ' )
144149
145150 if self .uarch == 'gh200' :
151+ self .executable = './mps-wrapper.sh -- ' + self .executable
152+
146153 self .executable_opts += [
147154 '-pme gpu' , '-nb gpu' , '-update gpu' , '-nsteps 10000'
148155 ]
0 commit comments