From 5d6a8838497c65e72c9b3a4209efab73240f1578 Mon Sep 17 00:00:00 2001
From: Bixia Zheng <bixia@google.com>
Date: Wed, 12 Feb 2020 14:46:17 -0800
Subject: [PATCH] [TF:TensorRT] Avoid generating TensorRT tests with XLA
 enabled.

PiperOrigin-RevId: 294762060
Change-Id: I4d76813645422c2b5ae342dee3ce9bcb9ce7ce90
---
 tensorflow/python/compiler/tensorrt/BUILD | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/tensorflow/python/compiler/tensorrt/BUILD b/tensorflow/python/compiler/tensorrt/BUILD
index ab448398948..a7c206f4495 100644
--- a/tensorflow/python/compiler/tensorrt/BUILD
+++ b/tensorflow/python/compiler/tensorrt/BUILD
@@ -6,6 +6,10 @@
 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
 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(
     default_visibility = ["//visibility:public"],
     licenses = ["notice"],  # Apache 2.0
@@ -93,6 +97,7 @@ cuda_py_test(
         "no_windows",
         "nomac",
     ],
+    xla_enable_strict_auto_jit = False,
     deps = [
         ":trt_convert_py",
         "//tensorflow/python:client_testlib",
@@ -142,6 +147,7 @@ cuda_py_tests(
         "no_windows",
         "nomac",
     ],
+    xla_enable_strict_auto_jit = False,
     deps = [
         ":tf_trt_integration_test_base",
         "//tensorflow/python:client_testlib",
@@ -162,6 +168,7 @@ cuda_py_tests(
         "nomac",
         "notap",  # b/140261407
     ],
+    xla_enable_strict_auto_jit = False,
     deps = [
         ":tf_trt_integration_test_base",
         "//tensorflow/python:client_testlib",
@@ -187,6 +194,7 @@ cuda_py_test(
         "no_windows",
         "nomac",
     ],
+    xla_enable_strict_auto_jit = False,
     deps = [
         ":tf_trt_integration_test_base",
         "//tensorflow/python:client_testlib",