[WIP] DEBUG only {2023.06,2023a} PyTorch-bundle v2.1.2

easystacks/software.eessi.io/2023.06/eessi-2023.06-eb-4.9.2-2023a.yml

@@ -45,3 +45,4 @@ easyconfigs:
       options:
         # see https://github.com/easybuilders/easybuild-easyconfigs/pull/20880
         from-commit: bc6e08f89759b8b70166de5bfcb5056b9db8ec90
+  - PyTorch-bundle-2.1.2-foss-2023a.eb

eb_hooks.py

@@ -285,6 +285,30 @@ def parse_hook_qt5_check_qtwebengine_disable(ec, eprefix):
         raise EasyBuildError("Qt5-specific hook triggered for non-Qt5 easyconfig?!")
 
 
+def parse_hook_sentencepiece_disable_tcmalloc_aarch64(ec, eprefix):
+    """
+    Disable using TC_Malloc on 'aarch64' CPUs
+    """
+    cpu_target = get_eessi_envvar('EESSI_SOFTWARE_SUBDIR')
+    if ec.name == 'SentencePiece' and ec.version in ['0.2.0']:
+        if cpu_target in [CPU_TARGET_AARCH64_GENERIC, CPU_TARGET_NEOVERSE_N1, CPU_TARGET_NEOVERSE_V1]:
+            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:
+            print_msg("parse_hook for SentencePiece on %s -> leaving configopts unchanged", cpu_target)
+    else:
+        raise EasyBuildError("SentencePiece-specific hook triggered for non-SentencePiece easyconfig?!")
+
+
 def parse_hook_ucx_eprefix(ec, eprefix):
     """Make UCX aware of compatibility layer via additional configuration options."""
     if ec.name == 'UCX':
@@ -762,6 +786,7 @@ def inject_gpu_property(ec):
     'OpenBLAS': parse_hook_openblas_relax_lapack_tests_num_errors,
     'pybind11': parse_hook_pybind11_replace_catch2,
     'Qt5': parse_hook_qt5_check_qtwebengine_disable,
+    'SentencePiece': parse_hook_sentencepiece_disable_tcmalloc_aarch64,
     'UCX': parse_hook_ucx_eprefix,
 }