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

adds toggle for using core masks

parent c9d3514b
No related branches found
No related tags found
No related merge requests found
Pipeline #122412 passed
......@@ -760,6 +760,9 @@ class _code(_base):
script_file.write('export OMP_NUM_THREADS=${SLURM_CPUS_PER_TASK}\n')
# explicit binding options that can be used further on if needed
if 'use_TurTLE_core_distribution' not in host_info.keys():
host_info['use_TurTLE_core_distribution'] = False
if host_info['use_TurTLE_core_distribution']:
core_masks = TurTLE.tools.distribute_cores_evenly(
nprocesses = nb_processes_per_node,
nthreads_per_process = nb_threads_per_process,
......
......@@ -34,7 +34,8 @@ host_info = {'type' : 'pc'}
# 'account' : info_template_account,
# 'executable_launcher' : info_template_executable_launcher,
# 'extra_slurm_lines' : info_template_extra_slurm_lines,
# 'explicit_slurm_environment' : info_template_explicit_slurm_environment}
# 'explicit_slurm_environment' : info_template_explicit_slurm_environment,
# 'use_TurTLE_core_distribution' : info_use_TurTLE_core_distribution}
# info_template_type can be one of:
# 'pc' --- jobs run interactively
......@@ -73,3 +74,7 @@ host_info = {'type' : 'pc'}
# in the submission script, or whether the cluster chooses it based on the
# resources requested
# info_use_TurTLE_core_distribution, relevant for `SLURM` clusters,
# is a bool specifying whether TurTLE should request explicit core
# masks for the individual MPI processes.
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment