Skip to content

Commit aad45d0

Browse files
committed
Merge branch 'LAMMPS_hook' into test_LAMMPS_HOOK
2 parents 5ad69ca + d935d9c commit aad45d0

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

eb_hooks.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1027,8 +1027,13 @@ def pre_configure_hook_LAMMPS_zen4_and_Aarch64_cuda(self, *args, **kwargs):
10271027
self.cfg['kokkos_arch'] = 'ZEN3'
10281028
elif get_cpu_architecture() == AARCH64:
10291029
if cpu_target == CPU_TARGET_NVIDIA_GRACE:
1030-
# There is no support for NVIDA grace in LAMMPS yet so falling back to ARMV81
1031-
self.cfg['kokkos_arch'] = 'ARMV81'
1030+
# There is no support for NVIDA grace in LAMMPS yet so falling back to older kokkos_arch
1031+
if self.cuda:
1032+
# Need to fall back to ARMV7 for CUDA builds because of problem with cuda
1033+
self.cfg['kokkos_arch'] = 'ARMV7'
1034+
else:
1035+
# There is no support for NVIDA grace in LAMMPS yet so falling back to ARMV81
1036+
self.cfg['kokkos_arch'] = 'ARMV81'
10321037

10331038
# Disable SIMD for specific CUDA versions
10341039
if self.version == '2Aug2023_update2':

0 commit comments

Comments
 (0)