From 487a8d7fb2fba20a72ccae62d67d4a3f8d84a52f Mon Sep 17 00:00:00 2001
From: Roy Frostig <frostig@google.com>
Date: Wed, 7 Nov 2018 14:53:20 -0800
Subject: [PATCH] [XLA] Track an existing SWIG include as a dependency in
 BUILD.

PiperOrigin-RevId: 220534068
---
 tensorflow/compiler/xla/python/BUILD | 1 +
 tensorflow/python/BUILD              | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/tensorflow/compiler/xla/python/BUILD b/tensorflow/compiler/xla/python/BUILD
index 21685c4a5b9..79603f21404 100644
--- a/tensorflow/compiler/xla/python/BUILD
+++ b/tensorflow/compiler/xla/python/BUILD
@@ -81,6 +81,7 @@ tf_py_wrap_cc(
     srcs = ["xla.i"],
     swig_includes = [
         "local_computation_builder.i",
+        "//tensorflow/python:platform/base.i",
     ],
     deps = [
         ":local_computation_builder",
diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD
index 7ff358cb088..bb9a9f6386c 100644
--- a/tensorflow/python/BUILD
+++ b/tensorflow/python/BUILD
@@ -20,6 +20,8 @@ licenses(["notice"])  # Apache 2.0
 
 exports_files(["LICENSE"])
 
+exports_files(["platform/base.i"])
+
 load("//tensorflow:tensorflow.bzl", "if_not_windows")
 load("//tensorflow:tensorflow.bzl", "tf_cuda_library")
 load("//tensorflow:tensorflow.bzl", "tf_gen_op_wrapper_py")