@@ -652,7 +652,7 @@ def parse_hook_maturin(ec, eprefix):
652652
653653def parse_hook_tensorflow_h5py_glibc (ec , eprefix ):
654654 """
655- Fix the Python and environment used while building and running tests for TensorFlow with CUDA
655+ Fix the Python and environment used while building and running tests for TensorFlow-2.18.1
656656 """
657657 cpu_target = get_eessi_envvar ('EESSI_SOFTWARE_SUBDIR' )
658658 if ec .name == 'TensorFlow' and ec .version == '2.18.1' :
@@ -677,18 +677,17 @@ def parse_hook_tensorflow_h5py_glibc(ec, eprefix):
677677 current_opts = current_opts .split ()
678678
679679 ec ['buildopts' ] = current_opts + [
680- '--linkopt=-Wl,--disable-new-dtags --host_linkopt=-Wl,--disable-new-dtags --action_env=GCC_HOST_COMPILER_PATH=$EBROOTGCC/bin/gcc --host_action_env=GCC_HOST_COMPILER_PATH=$EBROOTGCC/bin/gcc --linkopt=-Wl,-rpath,$EBROOTCUDA/lib:$EBROOTCUDNN/lib:$EBROOTNCCL/lib --host_linkopt=-Wl,-rpath,$EBROOTCUDA/lib:$EBROOTCUDNN/lib:$EBROOTNCCL/lib ' ,
680+ '--linkopt=-Wl,--disable-new-dtags --host_linkopt=-Wl,--disable-new-dtags --action_env=GCC_HOST_COMPILER_PATH=$EBROOTGCC/bin/gcc --host_action_env=GCC_HOST_COMPILER_PATH=$EBROOTGCC/bin/gcc ' ,
681681 ]
682682
683683 ec ['pretestopts' ] = (
684684 """interppath=$(find "$EESSI_EPREFIX/lib64" -name 'ld-*' | grep -E 'so\\ .1|so\\ .2' | head -n1) && """
685685 """pybin=$(find "%(builddir)s/%(name)s/bazel-root/" -type f -path "*/external/python_%(arch)s-unknown-linux-gnu/bin/python%(pyshortver)s" | head -n1) && """
686686 """patchelf --set-interpreter "$interppath" "$pybin" && """
687- """export LD_LIBRARY_PATH="$EBROOTCUDA/lib:$EBROOTCUDNN/lib:$EBROOTNCCL/lib:$LD_LIBRARY_PATH" && """
688687 )
689- print_msg ("TensorFlow-glibc related changes have been applied" )
688+ print_msg ("TensorFlow-h5py- glibc related changes have been applied" )
690689 else :
691- raise EasyBuildError ("TensorFlow-glibc specific hook triggered for non-TensorFlow-glibc easyconfig?!" )
690+ raise EasyBuildError ("TensorFlow-h5py- glibc specific hook triggered for non-TensorFlow easyconfig?!" )
692691
693692
694693def parse_hook_ucx_eprefix (ec , eprefix ):
0 commit comments