Merge pull request #1086 from trz42/2023.06-a64fx-2023a-eb491-apps-pytorch

{2023.06}[2023a,a64fx] PyTorch 2.1.2 originally built with EB 4.9.0

Author: bedroge

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:

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),