diff --git a/easystacks/software.eessi.io/2025.06/accel/amd/eessi-2025.06-eb-5.3.1-2025a.yml b/easystacks/software.eessi.io/2025.06/accel/amd/eessi-2025.06-eb-5.3.1-2025a.yml new file mode 100644 index 00000000..65560472 --- /dev/null +++ b/easystacks/software.eessi.io/2025.06/accel/amd/eessi-2025.06-eb-5.3.1-2025a.yml @@ -0,0 +1,2 @@ +easyconfigs: + - rocBLAS-4.4.0-rocm-compilers-19.0.0-ROCm-6.4.1.eb diff --git a/eb_hooks.py b/eb_hooks.py index 67b031bc..b641f75c 100644 --- a/eb_hooks.py +++ b/eb_hooks.py @@ -587,6 +587,13 @@ def parse_hook_cgal_toolchainopts_precise(ec, eprefix): raise EasyBuildError("CGAL-specific hook triggered for non-CGAL easyconfig?!") +def parse_hook_hipblaslt_single_core_build(ec, eprefix): + """Inject -DTensile_CPU_THREADS=1 to force Tensile to use a single core.""" + # todo: if this works, only do it for certain gfx targets + if ec.name == 'hipBLASLt': + ec.update('configopts', '-DTensile_CPU_THREADS=1') + + def parse_hook_fontconfig_add_fonts(ec, eprefix): """Inject --with-add-fonts configure option for fontconfig.""" if ec.name == 'fontconfig': @@ -2272,6 +2279,7 @@ def post_easyblock_hook(self, *args, **kwargs): 'fontconfig': parse_hook_fontconfig_add_fonts, 'FreeImage': parse_hook_freeimage_aarch64, 'grpcio': parse_hook_grpcio_zlib, + 'hipBLASLt': parse_hook_hipblaslt_single_core_build, 'maturin': parse_hook_maturin, 'Mesa': parse_hook_mesa_use_llvm_minimal, 'OpenBLAS': parse_hook_openblas_relax_lapack_tests_num_errors, @@ -2380,6 +2388,9 @@ def set_maximum(parallel, max_value): # Deucalion has 32GB HBM for 48 cores per node CPU_TARGET_A64FX: (divide_by_factor, 4), # software-specific limits + 'hipBLASLt': { + '*': (set_maximum, 1), #(divide_by_factor, 4), + }, 'libxc': { '*': (divide_by_factor, 2), CPU_TARGET_A64FX: (set_maximum, 6),