use 8 cores for PyTorch on A64FX

bedroge · bedroge · commit fa76fc6d29e7 · 2026-03-15T08:45:27.000+01:00
diff --git a/eb_hooks.py b/eb_hooks.py
@@ -2023,6 +2023,7 @@ def set_maximum(parallel, max_value):
     },
     'PyTorch': {
         CPU_TARGET_AARCH64_GENERIC: (divide_by_factor, 2),
+        CPU_TARGET_A64FX: (set_maximum, 8),
         CPU_TARGET_NEOVERSE_N1: (divide_by_factor, 2),
         CPU_TARGET_NEOVERSE_V1: (divide_by_factor, 2),
     },
