diff --git a/easystacks/software.eessi.io/2023.06/a64fx/eessi-2023.06-eb-4.9.4-2023a.yml b/easystacks/software.eessi.io/2023.06/a64fx/eessi-2023.06-eb-4.9.4-2023a.yml index 66b261ac9b..afc595ce24 100644 --- a/easystacks/software.eessi.io/2023.06/a64fx/eessi-2023.06-eb-4.9.4-2023a.yml +++ b/easystacks/software.eessi.io/2023.06/a64fx/eessi-2023.06-eb-4.9.4-2023a.yml @@ -170,12 +170,12 @@ easyconfigs: # # see https://github.com/easybuilders/easybuild-easyblocks/pull/3036 # include-easyblocks-from-pr: 3036 - LAMMPS-2Aug2023_update2-foss-2023a-kokkos.eb -## PR 19573 was included since EB 4.9.1 -## - PyTorch-2.1.2-foss-2023a.eb: -## options: -## # see https://github.com/easybuilders/easybuild-easyconfigs/pull/19573 -## from-pr: 19573 -# - PyTorch-2.1.2-foss-2023a.eb +# PR 19573 was included since EB 4.9.1 +# - PyTorch-2.1.2-foss-2023a.eb: +# options: +# # see https://github.com/easybuilders/easybuild-easyconfigs/pull/19573 +# from-pr: 19573 + - PyTorch-2.1.2-foss-2023a.eb - matplotlib-3.7.2-gfbf-2023a.eb # PR 19554 was included since EB 4.9.1 # - PyQt5-5.15.10-GCCcore-12.3.0.eb: diff --git a/eb_hooks.py b/eb_hooks.py index d92f457e41..211e24c9d0 100644 --- a/eb_hooks.py +++ b/eb_hooks.py @@ -1286,6 +1286,9 @@ def set_maximum(parallel, max_value): 'MBX': { '*': (divide_by_factor, 2), }, + 'PyTorch': { + CPU_TARGET_A64FX: (divide_by_factor, 4), + }, 'TensorFlow': { '*': (divide_by_factor, 2), CPU_TARGET_A64FX: (set_maximum, 8),