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

[TF:TensorRT] Avoid generating TensorRT tests with XLA enabled.

Browse Source 
PiperOrigin-RevId: 294762060
Change-Id: I4d76813645422c2b5ae342dee3ce9bcb9ce7ce90
This commit is contained in:
Bixia Zheng 2020-02-12 14:46:17 -08:00 committed by TensorFlower Gardener
parent d2c0a8f130
commit 5d6a883849
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/python/compiler/tensorrt/BUILD
View File

@ -6,6 +6,10 @@
load("//tensorflow:tensorflow.bzl", "cuda_py_test") load("//tensorflow:tensorflow.bzl", "cuda_py_test")
load("//tensorflow:tensorflow.bzl", "cuda_py_tests") load("//tensorflow:tensorflow.bzl", "cuda_py_tests")
# cuda_py_test and cuda_py_tests enable XLA tests by default. We can't
# combine XLA with TensorRT currently and should set
# xla_enable_strict_auto_jit to False to disable XLA tests.
package( package(
default_visibility = ["//visibility:public"], default_visibility = ["//visibility:public"],
licenses = ["notice"], # Apache 2.0 licenses = ["notice"], # Apache 2.0
@ -93,6 +97,7 @@ cuda_py_test(
"no_windows", "no_windows",
"nomac", "nomac",
], ],
xla_enable_strict_auto_jit = False,
deps = [ deps = [
":trt_convert_py", ":trt_convert_py",
"//tensorflow/python:client_testlib", "//tensorflow/python:client_testlib",
@ -142,6 +147,7 @@ cuda_py_tests(
"no_windows", "no_windows",
"nomac", "nomac",
], ],
xla_enable_strict_auto_jit = False,
deps = [ deps = [
":tf_trt_integration_test_base", ":tf_trt_integration_test_base",
"//tensorflow/python:client_testlib", "//tensorflow/python:client_testlib",
@ -162,6 +168,7 @@ cuda_py_tests(
"nomac", "nomac",
"notap", # b/140261407 "notap", # b/140261407
], ],
xla_enable_strict_auto_jit = False,
deps = [ deps = [
":tf_trt_integration_test_base", ":tf_trt_integration_test_base",
"//tensorflow/python:client_testlib", "//tensorflow/python:client_testlib",
@ -187,6 +194,7 @@ cuda_py_test(
"no_windows", "no_windows",
"nomac", "nomac",
], ],
xla_enable_strict_auto_jit = False,
deps = [ deps = [
":tf_trt_integration_test_base", ":tf_trt_integration_test_base",
"//tensorflow/python:client_testlib", "//tensorflow/python:client_testlib",