Skip to content

Commit 0c8e734

Browse files
Added logging info for mpi calculations
1 parent a66c23a commit 0c8e734

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

GMXMMPBSA/main.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -219,7 +219,10 @@ def load_calc_list(self):
219219
nframes = self.numframes if self.master else 0
220220
nmframes = self.numframes_nmode if self.master else 0
221221
self.calc_list = CalculationList(self.timer, nframes, nmframes, self.mpi_size)
222-
222+
if self.master:
223+
logging.info(f'Starting calculations in {self.mpi_size} CPUs...')
224+
if (self.INPUT['pbrun'] or self.INPUT['rismrun'] or self.INPUT['nmoderun']) and self.mpi_size > 1:
225+
logging.warning('PB/RISM/NMODE will be calculated with multiple threads, make sure you have enough RAM.')
223226
if not self.INPUT['mutant_only']:
224227
self.calc_list.append(PrintCalc('Running calculations on normal system...'), timer_key=None)
225228
self._load_calc_list(self.pre, False, self.normal_system)
@@ -712,6 +715,8 @@ def get_cl_args(self, args=None):
712715
args.remove('mpi')
713716
elif 'MPI' in args:
714717
args.remove('MPI')
718+
elif self.mpi_size > 1:
719+
pass
715720
else:
716721
_mpi = False
717722
mpi_cl = f' mpirun -np {self.mpi_size} ' if _mpi else ' '

0 commit comments

Comments
 (0)