Skip to content

Commit

Permalink
Merge pull request #2841 from Flamefire/20221202124223_new_pr_tensorflow
Browse files Browse the repository at this point in the history
Make TensorFlow builds easier to debug
  • Loading branch information
branfosj authored Dec 4, 2022
2 parents a01e2a8 + 0d7b232 commit 99ee865
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions easybuild/easyblocks/t/tensorflow.py
Original file line number Diff line number Diff line change
Expand Up @@ -430,10 +430,14 @@ def get_system_libs(self):

def setup_build_dirs(self):
"""Setup temporary build directories"""
# This is either the builddir (for standalone builds) or the extension sub folder when TF is an extension
# Either way this folder only contains the folder with the sources and hence we can use fixed names
# for the subfolders
parent_dir = os.path.dirname(self.start_dir)
# Path where Bazel will store its output, build artefacts etc.
self.output_user_root_dir = tempfile.mkdtemp(suffix='-bazel-tf', dir=self.builddir)
self.output_user_root_dir = os.path.join(parent_dir, 'bazel-root')
# Folder where wrapper binaries can be placed, where required. TODO: Replace by --action_env cmds
self.wrapper_dir = tempfile.mkdtemp(suffix='-wrapper_bin', dir=self.builddir)
self.wrapper_dir = os.path.join(parent_dir, 'wrapper_bin')

def configure_step(self):
"""Custom configuration procedure for TensorFlow."""
Expand Down

0 comments on commit 99ee865

Please sign in to comment.