From 278ae61db464d60cf1717467b0d126a419afca1f Mon Sep 17 00:00:00 2001 From: Thomas Roeblitz Date: Tue, 11 Jun 2024 06:24:12 +0200 Subject: [PATCH] EBROOT* env vars aren't set in parse hooks --- eb_hooks.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/eb_hooks.py b/eb_hooks.py index 530b362123..3adf7cb98e 100644 --- a/eb_hooks.py +++ b/eb_hooks.py @@ -308,10 +308,12 @@ def parse_hook_pytorch_bundle_torchvision_setenv(ec, eprefix): print_msg("parse_hook for PyTorch-bundle without CUDA: extslist '%s'", ec['exts_list']) print_msg("New exts_list: '%s'", ec['exts_list']) ec_dict = ec.asdict() - libpng_root = os.getenv('EBROOTLIBPNG') + cpu_target = get_eessi_envvar('EESSI_SOFTWARE_SUBDIR') + eessi_software_path = get_eessi_envvar('EESSI_SOFTWARE_PATH') + libpng_root = os.path.join(eessi_software_path, "software", "libpng", "1.6.39-GCCcore-12.3.0") libpng_include = os.path.join(libpng_root, 'include') libpng_lib = os.path.join(libpng_root, 'lib') - libjpeg_turbo_root = os.getenv('EBROOTLIBJPEGMINTURBO') + libjpeg_turbo_root = os.path.join(eessi_software_path, "software", "libjpeg-turbo", "2.1.5.1-GCCcore-12.3.0") libjpeg_turbo_include = os.path.join(libjpeg_turbo_root, 'include') libjpeg_turbo_lib = os.path.join(libjpeg_turbo_root, 'lib') exts_list_new = []