Skip to content

Commit 28a3276

Browse files
committed
add version checks for symlinks in comm_libs folder of NVHPC
1 parent 9815b17 commit 28a3276

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

easybuild/easyblocks/generic/nvidiabase.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -480,11 +480,13 @@ def install_step(self):
480480
if not self.cfg['module_add_nccl']:
481481
nccl_dir_glob = os.path.join(abs_install_subdir, 'comm_libs', self.CUDA_VERSION_GLOB, 'nccl')
482482
remove(glob.glob(nccl_dir_glob))
483-
remove(os.path.join(abs_install_subdir, 'comm_libs', 'nccl'))
483+
if LooseVersion(self.version) >= LooseVersion('25.0'):
484+
remove(os.path.join(abs_install_subdir, 'comm_libs', 'nccl'))
484485
if not self.cfg['module_add_nvshmem']:
485486
shmem_dir_glob = os.path.join(abs_install_subdir, 'comm_libs', self.CUDA_VERSION_GLOB, 'nvshmem')
486487
remove(glob.glob(shmem_dir_glob))
487-
remove(os.path.join(abs_install_subdir, 'comm_libs', 'nvshmem'))
488+
if LooseVersion(self.version) >= LooseVersion('25.0'):
489+
remove(os.path.join(abs_install_subdir, 'comm_libs', 'nvshmem'))
488490
if not self.cfg['module_add_math_libs']:
489491
remove(os.path.join(abs_install_subdir, 'math_libs'))
490492
if not self.cfg['module_add_profilers']:

0 commit comments

Comments
 (0)