Skip to content
Snippets Groups Projects
Commit 0d970ca9 authored by Cristian Lalescu's avatar Cristian Lalescu
Browse files

make launcher more flexible

parent c730977e
No related branches found
No related tags found
No related merge requests found
Pipeline #75648 passed
......@@ -403,11 +403,16 @@ class _code(_base):
subprocess.check_call(submit_atoms + [os.path.join(self.work_dir, job_script_name)])
elif self.host_info['type'] == 'pc':
command_atoms = ['mpirun',
'-np',
'{0}'.format(nb_processes),
'./' + self.name,
self.simname]
if 'executable_launcher' in self.host_info.keys():
command_atoms = [self.host_info['executable_launcher']]
if self.host_info['executable_launcher'] == 'srun':
command_atoms += ['-p', 'interactive']
else:
command_atoms = ['mpirun']
command_atoms += ['-n',
'{0}'.format(nb_processes),
'./' + self.name,
self.simname]
os.chdir(self.work_dir)
custom_env = os.environ
custom_env['OMP_NUM_THREADS'] = str(nb_threads_per_process)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment