Skip to content

Commit 9c6129d

Browse files
committed
move setting kokkos_arch=ARMV70
Signed-off-by: laraPPr <[email protected]>
1 parent a6b7cfd commit 9c6129d

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

eb_hooks.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1020,6 +1020,7 @@ def pre_configure_hook_LAMMPS_zen4_and_aarch64_cuda(self, *args, **kwargs):
10201020
if self.name == 'LAMMPS':
10211021

10221022
# Set kokkos_arch for LAMMPS version which do not have support for the target architecture
1023+
# This is no longer required with easybuild 5.1.2
10231024
if self.version in ('2Aug2023_update2', '2Aug2023_update4', '29Aug2024'):
10241025
if get_cpu_architecture() == X86_64:
10251026
if cpu_target == CPU_TARGET_ZEN4:
@@ -1029,12 +1030,6 @@ def pre_configure_hook_LAMMPS_zen4_and_aarch64_cuda(self, *args, **kwargs):
10291030
if cpu_target == CPU_TARGET_NVIDIA_GRACE:
10301031
# There is no support for NVIDA grace in LAMMPS yet so falling back to ARMV81
10311032
self.cfg['kokkos_arch'] = 'ARMV81'
1032-
# To disable simd kokkos_arch need to be set to ARMV7
1033-
if ('CUDA' in [dep['name'] for dep in deps]):
1034-
for dep in deps:
1035-
if 'CUDA' == dep['name']:
1036-
if dep['version'] in cuda_versions:
1037-
self.cfg['kokkos_arch'] = 'ARMV70'
10381033

10391034
# Disable SIMD for specific CUDA versions
10401035
if self.version == '2Aug2023_update2':
@@ -1043,6 +1038,7 @@ def pre_configure_hook_LAMMPS_zen4_and_aarch64_cuda(self, *args, **kwargs):
10431038
for dep in deps:
10441039
if 'CUDA' == dep['name']:
10451040
if dep['version'] in cuda_versions:
1041+
self.cfg['kokkos_arch'] = 'ARMV70'
10461042
cxxflags = os.getenv('CXXFLAGS', '')
10471043
cxxflags = cxxflags.replace('-mcpu=native', '')
10481044
cxxflags += ' -march=armv8-a+nosimd'

0 commit comments

Comments
 (0)