set PyTorch CPU limits for aarch64/{generic,neoverse_n1,neoverse_v1}

Author: bedroge

eb_hooks.py

@@ -2021,6 +2021,11 @@ def set_maximum(parallel, max_value):
         '*': (divide_by_factor, 2),
         CPU_TARGET_A64FX: (set_maximum, 1),
     },
+    'PyTorch': {
+        CPU_TARGET_AARCH64_GENERIC: (divide_by_factor, 2),
+        CPU_TARGET_NEOVERSE_N1: (divide_by_factor, 2),
+        CPU_TARGET_NEOVERSE_V1: (divide_by_factor, 2),
+    },
     'QuantumESPRESSO': {
         CPU_TARGET_A64FX: (set_maximum, 6),
     },