Skip to content

Commit

Permalink
fix logic to add directory to NCCL libs
Browse files Browse the repository at this point in the history
  • Loading branch information
truib committed Oct 29, 2024
1 parent 13dd418 commit 0f331a8
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions easybuild/easyblocks/t/tensorflow.py
Original file line number Diff line number Diff line change
Expand Up @@ -565,9 +565,11 @@ def configure_step(self):
# add path to libnccl.so.2 directory provided by NCCL when both sysroot
# and RPATH are used (such as in EESSI)
if build_option('sysroot') and self.toolchain.use_rpath:
libpaths = self.system_libs_info[2]
libpaths.append(os.path.join(nccl_root, 'lib'))
self.system_libs_info[2] = libpaths
system_libs_info_as_list = list(self.system_libs_info)
new_libpaths = system_libs_info_as_list[2]
new_libpaths.append(os.path.join(nccl_root, 'lib'))
system_libs_info_as_list[2] = new_libpaths
self.system_libs_info = tuple(system_libs_info_as_list)

self._with_cuda = bool(cuda_root)

Expand Down

0 comments on commit 0f331a8

Please sign in to comment.