From da82af74a0f5fd700842cda0f847038ed9469c5e Mon Sep 17 00:00:00 2001 From: lara Date: Wed, 15 Nov 2023 09:49:29 +0100 Subject: [PATCH] change kokkos mapping for Aarch64" --- eb_hooks.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/eb_hooks.py b/eb_hooks.py index a2890da75e..dfd91ab18b 100644 --- a/eb_hooks.py +++ b/eb_hooks.py @@ -19,6 +19,7 @@ CPU_TARGET_NEOVERSE_V1 = 'aarch64/neoverse_v1' +CPU_TARGET_AARCH64_GENERIC = 'aarch64/generic' EESSI_RPATH_OVERRIDE_ATTR = 'orig_rpath_override_dirs' @@ -277,7 +278,10 @@ def pre_configure_hook_LAMMPS_aarch64(self, *args, **kwargs): if self.name == 'LAMMPS': if self.version == '23Jun2022': if get_cpu_architecture() == AARCH64: - self.cfg['kokkos_arch'] = 'A64FX' + if cpu_target == CPU_TARGET_AARCH64_GENERIC: + self.cfg['kokkos_arch'] = 'ARM81' + else: + self.cfg['kokkos_arch'] = 'ARM80' else: raise EasyBuildError("LAMMPS-specific hook triggered for non-LAMMPS easyconfig?!")