From b42a7a2045cbb3aed5b9c6fd49580b6fe74731fe Mon Sep 17 00:00:00 2001 From: laraPPr Date: Fri, 27 Jun 2025 16:04:42 +0200 Subject: [PATCH 1/3] update eb_hooks.py for memory expensive builds of jax Signed-off-by: laraPPr --- eb_hooks.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/eb_hooks.py b/eb_hooks.py index e20ef2e4..c5ded8da 100644 --- a/eb_hooks.py +++ b/eb_hooks.py @@ -1399,4 +1399,7 @@ def set_maximum(parallel, max_value): 'Qt5': { CPU_TARGET_A64FX: (set_maximum, 8), }, + 'jax': { + '*': (divide_by_factor, 2), + }, } From 77691177b67e9ea0a995dbc622b65c3867802f6e Mon Sep 17 00:00:00 2001 From: laraPPr Date: Fri, 27 Jun 2025 16:51:04 +0200 Subject: [PATCH 2/3] test eb_hooks.py with test easystack Signed-off-by: laraPPr --- .../2023.06/eessi-2023.06-eb-5.1.0-2023a.yml | 5 +++++ 1 file changed, 5 insertions(+) create mode 100644 easystacks/software.eessi.io/2023.06/eessi-2023.06-eb-5.1.0-2023a.yml diff --git a/easystacks/software.eessi.io/2023.06/eessi-2023.06-eb-5.1.0-2023a.yml b/easystacks/software.eessi.io/2023.06/eessi-2023.06-eb-5.1.0-2023a.yml new file mode 100644 index 00000000..2d7751f7 --- /dev/null +++ b/easystacks/software.eessi.io/2023.06/eessi-2023.06-eb-5.1.0-2023a.yml @@ -0,0 +1,5 @@ +easyconfigs: + - jax-0.4.25-gfbf-2023a.eb: + options: + # see https://github.com/easybuilders/easybuild-easyblocks/pull/3486 + include-easyblocks-from-commit: 9266abcd6c5668c5aca5815ea6c04ccf43112555 From fb79b5bf8d2f444622ccd43ed4e31f7d3572e691 Mon Sep 17 00:00:00 2001 From: Lara Ramona Peeters <49882639+laraPPr@users.noreply.github.com> Date: Mon, 30 Jun 2025 10:01:54 +0200 Subject: [PATCH 3/3] Use older commit of easyblock This commit was used in the original pr in the software-layer maybe this was for a reason. --- .../software.eessi.io/2023.06/eessi-2023.06-eb-5.1.0-2023a.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/easystacks/software.eessi.io/2023.06/eessi-2023.06-eb-5.1.0-2023a.yml b/easystacks/software.eessi.io/2023.06/eessi-2023.06-eb-5.1.0-2023a.yml index 2d7751f7..386a1e89 100644 --- a/easystacks/software.eessi.io/2023.06/eessi-2023.06-eb-5.1.0-2023a.yml +++ b/easystacks/software.eessi.io/2023.06/eessi-2023.06-eb-5.1.0-2023a.yml @@ -2,4 +2,4 @@ easyconfigs: - jax-0.4.25-gfbf-2023a.eb: options: # see https://github.com/easybuilders/easybuild-easyblocks/pull/3486 - include-easyblocks-from-commit: 9266abcd6c5668c5aca5815ea6c04ccf43112555 + include-easyblocks-from-commit: 59fd120c0e5330ff7266ca7aacd105d09be4389a