experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

third_party/repo: add system_link_files

Browse Source 
third_party_http_archive has a link_files attr, add a similar
system_link_files attr that is only used when the system libraries are
enabled. Also add it to tf_http_archive.

Signed-off-by: Jason Zaman <jason@perfinion.com>
This commit is contained in:
Jason Zaman 2018-09-06 22:01:10 +08:00
parent 05b43db45d
commit 37f7bfbce8
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
third_party/repo.bzl vendored
View File

@ -119,6 +119,10 @@ def _tf_http_archive(ctx):
"%prefix%": ".." if _repos_are_siblings() else "external", "%prefix%": ".." if _repos_are_siblings() else "external",
}, False) }, False)
if use_syslib:
for internal_src, external_dest in ctx.attr.system_link_files.items():
ctx.symlink(Label(internal_src), ctx.path(external_dest))
tf_http_archive = repository_rule( tf_http_archive = repository_rule(
implementation = _tf_http_archive, implementation = _tf_http_archive,
attrs = { attrs = {
@ -130,6 +134,7 @@ tf_http_archive = repository_rule(
"patch_file": attr.label(), "patch_file": attr.label(),
"build_file": attr.label(), "build_file": attr.label(),
"system_build_file": attr.label(), "system_build_file": attr.label(),
"system_link_files": attr.string_dict(),
}, },
environ = [ environ = [
"TF_SYSTEM_LIBS", "TF_SYSTEM_LIBS",
@ -180,7 +185,16 @@ def _third_party_http_archive(ctx):
_apply_patch(ctx, ctx.attr.patch_file) _apply_patch(ctx, ctx.attr.patch_file)
ctx.symlink(Label(ctx.attr.build_file), buildfile_path) ctx.symlink(Label(ctx.attr.build_file), buildfile_path)
link_dict = dict()
if use_syslib:
link_dict.update(ctx.attr.system_link_files)
for internal_src, external_dest in ctx.attr.link_files.items(): for internal_src, external_dest in ctx.attr.link_files.items():
# if syslib and link exists in both, use the system one
if external_dest not in link_dict.values():
link_dict[internal_src] = external_dest
for internal_src, external_dest in link_dict.items():
ctx.symlink(Label(internal_src), ctx.path(external_dest)) ctx.symlink(Label(internal_src), ctx.path(external_dest))
# Downloads and creates Bazel repos for dependencies. # Downloads and creates Bazel repos for dependencies.
@ -201,6 +215,7 @@ third_party_http_archive = repository_rule(
"system_build_file": attr.string(mandatory = False), "system_build_file": attr.string(mandatory = False),
"patch_file": attr.label(), "patch_file": attr.label(),
"link_files": attr.string_dict(), "link_files": attr.string_dict(),
"system_link_files": attr.string_dict(),
}, },
environ = [ environ = [
"TF_SYSTEM_LIBS", "TF_SYSTEM_LIBS",