Skip to content

Commit 0a48329

Browse files
committed
implement fix for all CUDA versions older than 13.1
Signed-off-by: laraPPr <[email protected]>
1 parent 9c6129d commit 0a48329

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

eb_hooks.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1013,10 +1013,6 @@ def pre_configure_hook_LAMMPS_zen4_and_aarch64_cuda(self, *args, **kwargs):
10131013
# Get cpu_target for zen4 hook
10141014
cpu_target = get_eessi_envvar('EESSI_SOFTWARE_SUBDIR')
10151015

1016-
# Get the dependencies to check CUDA
1017-
deps = self.cfg.dependencies()
1018-
cuda_versions = ('12.1.1')
1019-
10201016
if self.name == 'LAMMPS':
10211017

10221018
# Set kokkos_arch for LAMMPS version which do not have support for the target architecture
@@ -1034,10 +1030,11 @@ def pre_configure_hook_LAMMPS_zen4_and_aarch64_cuda(self, *args, **kwargs):
10341030
# Disable SIMD for specific CUDA versions
10351031
if self.version == '2Aug2023_update2':
10361032
if get_cpu_architecture() == AARCH64:
1037-
if ('CUDA' in [dep['name'] for dep in deps]):
1038-
for dep in deps:
1033+
if self.cuda:
1034+
for dep in deps = self.cfg.dependencies():
10391035
if 'CUDA' == dep['name']:
1040-
if dep['version'] in cuda_versions:
1036+
# This was broken until CUDA 13.1
1037+
if dep['version'] < LooseVersion('13.1'):
10411038
self.cfg['kokkos_arch'] = 'ARMV70'
10421039
cxxflags = os.getenv('CXXFLAGS', '')
10431040
cxxflags = cxxflags.replace('-mcpu=native', '')

0 commit comments

Comments
 (0)