@@ -279,29 +279,23 @@ def parse_hook_tensorflow_CUDA(ec, eprefix):
279279 )
280280
281281 ec ['buildopts' ] = [
282- '- - linkopt = - Wl ,- - disable - new - dtags - - host_linkopt = - Wl ,- - disable - new - dtags - - action_env = GCC_HOST_COMPILER_PATH = $EBROOT
283- GCC / bin / gcc - - host_action_env = GCC_HOST_COMPILER_PATH = $EBROOTGCC / bin / gcc - - linkopt = - Wl ,- rpath ,$EBROOTCUDA / lib :$EBROOTCUDNN / lib :$EBROOT
284- NCCL / lib - - host_linkopt = - Wl ,- rpath ,$EBROOTCUDA / lib :$EBROOTCUDNN / lib :$EBROOTNCCL / lib ',
282+ '--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' ,
285283 ]
286284
287285 ec ['pretestopts' ] = (
288- #'patchelf --set-interpreter "$EESSI_EPREFIX/lib64/ld-linux-%(arch)s.so.1" '
289- #'"%(builddir)s/%(name)s/bazel-root/0b9648e0837f9e5bb579e0e2e64adf3f/external/python_aarch64-unknown-linux-gnu/bin/python%(pyshortver)s" && '
290-
291- 'mv "%(builddir)s/%(name)s/bazel-root/0b9648e0837f9e5bb579e0e2e64adf3f/external/python_aarch64-unknown-linux-gnu/bin/python%(pyshortver)s" "%(builddir)s/%(name)s/bazel-root/0b9648e0837f9e5bb579e0e2e64adf3f/external/python_aarch64-unknown-linux-gnu/bin/python%(pyshortver)s.old" && cp -f $EBROOTPYTHON/bin/python%(pyshortver)s "%(builddir)s/%(name)s/bazel-root/0b9648e0837f9e5bb579e0e2e64adf3f/external/python_aarch64-unknown-linux-gnu/bin/python%(pyshortver)s" && '
292-
293-
294-
295- #'LD_LIBRARY_PATH=$EBROOTCUDA/lib:$EBROOTCUDNN/lib:$EBROOTNCCL/lib:$LD_LIBRARY_PATH && '
296- )
286+ 'interppath=$(find "$EESSI_EPREFIX/lib64" -name \' ld-*\' | grep -E \' so\\ .1|so\\ .2\' | head -n1) && '
287+ 'patchelf --set-interpreter "$interppath" '
288+ '"%(builddir)s/%(name)s/bazel-root/0b9648e0837f9e5bb579e0e2e64adf3f/external/python_%(arch)s-unknown-linux-gnu/bin/python%(pyshortver)s" && '
289+ 'export LD_LIBRARY_PATH="$EBROOTCUDA/lib:$EBROOTCUDNN/lib:$EBROOTNCCL/lib:$LD_LIBRARY_PATH" && '
290+ )
297291
298292 ec ['postinstallcmds' ] = [
299293 'mkdir -p %(installdir)s/bin' ,
300294 'ln -s $EBROOTCUDA/bin/cuobjdump %(installdir)s/bin/cuobjdump' ,
301- 'chmod 755 -R %(builddir)s' ,
295+ # 'chmod 755 -R %(builddir)s',
302296 ]
303297
304- print_msg ("TensorFlow-CUDA required changes are applied!!!" ),
298+ print_msg ("TensorFlow-CUDA related changes have been applied" )
305299 else :
306300 raise EasyBuildError ("TensorFlow-CUDA specific hook triggered for non-TensorFlow-CUDA easyconfig?!" )
307301
@@ -325,7 +319,7 @@ def parse_hook_casacore_disable_vectorize(ec, eprefix):
325319 if 'toolchainopts' not in ec or ec ['toolchainopts' ] is None :
326320 ec ['toolchainopts' ] = {}
327321 ec ['toolchainopts' ]['vectorize' ] = False
328- print_msg ("Changed toochainopts for %s: %s" , ec .name , ec ['toolchainopts' ])
322+ print_msg ("Changed toolchainopts for %s: %s" , ec .name , ec ['toolchainopts' ])
329323 else :
330324 print_msg ("Not changing option vectorize for %s on non-neoverse_v1" , ec .name )
331325 else :
0 commit comments