Skip to content

Commit a6b7cfd

Browse files
committed
restrict the use of setting ARMV70
Signed-off-by: laraPPr <[email protected]>
1 parent c2911b8 commit a6b7cfd

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

eb_hooks.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1019,7 +1019,7 @@ def pre_configure_hook_LAMMPS_zen4_and_aarch64_cuda(self, *args, **kwargs):
10191019

10201020
if self.name == 'LAMMPS':
10211021

1022-
# Set kokkos_arch for LAMMPS version which do not have support for versions that do not support ZEN4
1022+
# Set kokkos_arch for LAMMPS version which do not have support for the target architecture
10231023
if self.version in ('2Aug2023_update2', '2Aug2023_update4', '29Aug2024'):
10241024
if get_cpu_architecture() == X86_64:
10251025
if cpu_target == CPU_TARGET_ZEN4:
@@ -1030,8 +1030,11 @@ def pre_configure_hook_LAMMPS_zen4_and_aarch64_cuda(self, *args, **kwargs):
10301030
# There is no support for NVIDA grace in LAMMPS yet so falling back to ARMV81
10311031
self.cfg['kokkos_arch'] = 'ARMV81'
10321032
# To disable simd kokkos_arch need to be set to ARMV7
1033-
if self.cuda:
1034-
self.cfg['kokkos_arch'] = 'ARMV70'
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'
10351038

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

0 commit comments

Comments
 (0)