From f6c02908ec40cf64b6f30e8fbe420a9602e327f9 Mon Sep 17 00:00:00 2001 From: Thomas Roeblitz Date: Fri, 7 Jun 2024 12:11:33 +0200 Subject: [PATCH] add config opt to disable TC_MALLOC --- eb_hooks.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/eb_hooks.py b/eb_hooks.py index 6460cf2d77..e7999ddb16 100644 --- a/eb_hooks.py +++ b/eb_hooks.py @@ -319,7 +319,17 @@ def parse_hook_sentencepiece_disable_tcmalloc_aarch64(ec, eprefix): cpu_target = get_eessi_envvar('EESSI_SOFTWARE_SUBDIR') if ec.name == 'SentencePiece' and ec.version in ['0.2.0'] and cpu_target == CPU_TARGET_AARCH64_GENERIC: # find right setting to change/update - print_msg("parse_hook for SentencePiece: '%s'",ec['components']) + print_msg("parse_hook for SentencePiece: OLD '%s'", ec['components']) + new_components = [] + for item in ec['components']: + if item[2]['easyblock'] == 'CMakeMake': + new_item = item[2] + new_item['configopts'] = '-DSPM_ENABLE_TCMALLOC=OFF' + new_components.append((item[0], item[1], new_item)) + else: + new_components.append(item) + ec['components'] = new_components + print_msg("parse_hook for SentencePiece: NEW '%s'", ec['components']) else: raise EasyBuildError("SentencePiece-specific hook triggered for non-SentencePiece easyconfig?!")