Skip to content

Commit cae1c39

Browse files
authored
Merge pull request #3943 from laraPPr/LAMMPS_GENERIC_ARM
Make it possible to overide kokkos_arch for GENERIC ARM target
2 parents 61b8cf5 + 338c317 commit cae1c39

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

easybuild/easyblocks/l/lammps.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -276,7 +276,16 @@ def get_kokkos_arch(self, cuda_cc, kokkos_arch):
276276
# For other architectures we set a custom/non-existent type, which will disable all optimizations,
277277
# and it should use the compiler (optimization) flags set by EasyBuild for this architecture.
278278
if get_cpu_architecture() == AARCH64:
279-
processor_arch = 'ARMV80'
279+
if kokkos_arch:
280+
# If someone is trying a manual override for this case, let them
281+
if kokkos_arch not in KOKKOS_CPU_ARCH_LIST:
282+
warning_msg = "Specified CPU ARCH (%s) " % kokkos_arch
283+
warning_msg += "was not found in listed options [%s]." % KOKKOS_CPU_ARCH_LIST
284+
warning_msg += "Still might work though."
285+
print_warning(warning_msg)
286+
processor_arch = kokkos_arch
287+
else:
288+
processor_arch = 'ARMV80'
280289
else:
281290
processor_arch = 'EASYBUILD_GENERIC'
282291

0 commit comments

Comments
 (0)