diff --git a/TurTLE/_code.py b/TurTLE/_code.py index 54dca56f1fc5a4dc9a3f404f2fe67b933d109774..6e15f36892acb716d11d7798f5533cf0076c0e15 100644 --- a/TurTLE/_code.py +++ b/TurTLE/_code.py @@ -766,9 +766,13 @@ class _code(_base): else: executable_launcher = 'srun' if host_info['use_vtune']: - executable_launcher = 'srun vtune -collect hpc-performance -trace-mpi -quiet -result-dir=vtune_${SLURM_JOB_NAME}' + if 'vtune_executable' not in host_info.keys(): + host_info['vtune_executable'] = 'vtune' + executable_launcher = 'srun ' + host_info['vtune_executable'] + ' -collect hpc-performance -trace-mpi -quiet -result-dir=vtune_${SLURM_JOB_NAME}' if host_info['use_aps']: - executable_launcher = 'srun aps --result-dir=aps_${SLURM_JOB_NAME} --collection-mode=all' + if 'aps_executable' not in host_info.keys(): + host_info['aps_executable'] = 'aps' + executable_launcher = 'srun ' + host_info['aps_executable'] + ' --result-dir=aps_${SLURM_JOB_NAME} --collection-mode=all' script_file.write(executable_launcher + ' {0}\n'.format(' '.join(command_atoms))) if host_info['use_aps']: script_file.write('aps-report aps_${SLURM_JOB_NAME}\n')