Skip to content

Commit 5befb75

Browse files
author
Richard Top
committed
fixed eb_hooks.py
1 parent 9f68c5b commit 5befb75

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

eb_hooks.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -255,7 +255,6 @@ def post_prepare_hook(self, *args, **kwargs):
255255
if cpu_target == CPU_TARGET_ZEN4:
256256
post_prepare_hook_ignore_zen4_gcccore1220_error(self, *args, **kwargs)
257257

258-
259258
def parse_hook_tensorflow_CUDA(ec, eprefix):
260259
"""
261260
Fix the Python and environment used while building and running tests for TensorFlow with CUDA
@@ -279,8 +278,7 @@ def parse_hook_tensorflow_CUDA(ec, eprefix):
279278
)
280279

281280
ec['buildopts'] = [
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:$EBROOT
283-
NCCL/lib --host_linkopt=-Wl,-rpath,$EBROOTCUDA/lib:$EBROOTCUDNN/lib:$EBROOTNCCL/lib',
281+
'--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',
284282
]
285283

286284
ec['pretestopts'] = (
@@ -292,10 +290,10 @@ def parse_hook_tensorflow_CUDA(ec, eprefix):
292290
ec['postinstallcmds'] = [
293291
'mkdir -p %(installdir)s/bin',
294292
'ln -s $EBROOTCUDA/bin/cuobjdump %(installdir)s/bin/cuobjdump',
295-
'chmod 755 -R %(builddir)s',
293+
#'chmod 755 -R %(builddir)s',
296294
]
297295

298-
print_msg("TensorFlow-CUDA required changes are applied!!!"),
296+
print_msg("TensorFlow-CUDA required changes are applied!!!")
299297
else:
300298
raise EasyBuildError("TensorFlow-CUDA specific hook triggered for non-TensorFlow-CUDA easyconfig?!")
301299

0 commit comments

Comments
 (0)