Skip to content

Commit 8c78f2f

Browse files
committed
Merge branch 'LAMMPS_hook' into test_LAMMPS_HOOK
2 parents aad45d0 + 6563669 commit 8c78f2f

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

eb_hooks.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1027,13 +1027,11 @@ 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 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'
1030+
# There is no support for NVIDA grace in LAMMPS yet so falling back to ARMV81
1031+
self.cfg['kokkos_arch'] = 'ARMV81'
1032+
# To disable simd kokkos_arch need to be set to ARMV7
1033+
if self.cuda:
1034+
self.cfg['kokkos_arch'] = 'ARMV7'
10371035

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

0 commit comments

Comments
 (0)