tweak libpaths by adding directory containing libnccl.so.2

easybuilders · Oct 28, 2024 · 13dd418 · 13dd418
1 parent 3671c5b
commit 13dd418
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/easybuild/easyblocks/t/tensorflow.py b/easybuild/easyblocks/t/tensorflow.py
@@ -562,6 +562,13 @@ def configure_step(self):
         tensorrt_root = get_software_root('TensorRT')
         nccl_root = get_software_root('NCCL')
 
+        # 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
+
         self._with_cuda = bool(cuda_root)
 
         config_env_vars = {