change kokkos mapping for Aarch64"

EESSI · Nov 15, 2023 · da82af7 · da82af7
1 parent fd6ac5a
commit da82af7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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?!")