Skip to content

Commit d935d9c

Browse files
committed
fall back to ARMV7 for LAMMPS build with CUDA
Signed-off-by: laraPPr <[email protected]>
1 parent 9bd9e9a commit d935d9c

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)