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

Fix TRT config that is used when building GPU package without TRT.

Browse Source 
PiperOrigin-RevId: 246023930
This commit is contained in:
Guangda Lai 2019-04-30 14:38:57 -07:00 committed by TensorFlower Gardener
parent 52c7b0d597
commit bb5a94d15a
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
third_party/tensorrt/tensorrt_configure.bzl vendored
View File

@ -37,11 +37,10 @@ def _tpl(repository_ctx, tpl, substitutions):
def _create_dummy_repository(repository_ctx):
"""Create a dummy TensorRT repository."""
_tpl(repository_ctx, "build_defs.bzl", {"%{if_tensorrt}": "if_false"})
_tpl(repository_ctx, "BUILD", {
"%{tensorrt_genrules}": "",
"%{tensorrt_headers}": "[]",
"%{tensorrt_libs}": "[]",
"%{copy_rules}": "",
"\":tensorrt_include\"": "",
"\":tensorrt_lib\"": "",
})
def enable_tensorrt(repository_ctx):
@ -95,7 +94,6 @@ def _tensorrt_configure_impl(repository_ctx):
# Set up BUILD file.
_tpl(repository_ctx, "BUILD", {
"%{copy_rules}": "\n".join(copy_rules),
"%{tensorrt_libs}": str(libraries),
})
tensorrt_configure = repository_rule(