Skip to content

Commit 6451f30

Browse files
author
Richard Top
committed
removed CUDA entries
1 parent d510e7e commit 6451f30

1 file changed

Lines changed: 4 additions & 5 deletions

File tree

eb_hooks.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -652,7 +652,7 @@ def parse_hook_maturin(ec, eprefix):
652652

653653
def 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

694693
def parse_hook_ucx_eprefix(ec, eprefix):

0 commit comments

Comments
 (0)