1 files changed, 39 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/BUILD b/tensorflow/stream_executor/BUILD
new file mode 100644
index 0000000000..b91fe431f6
--- /dev/null
+++ b/tensorflow/stream_executor/BUILD
@@ -0,0 +1,39 @@
+licenses(["restricted"])
+
+load("/tensorflow/tensorflow", "if_cuda")
+
+cc_library(
+    name = "stream_executor",
+    srcs = glob(
+        [
+            "*.cc",
+            "lib/*.cc",
+        ],
+        exclude = [
+            "**/*_test.cc",
+        ],
+    ) + if_cuda(
+        glob([
+            "cuda/*.cc",
+        ]),
+    ),
+    hdrs = glob([
+        "*.h",
+        "lib/*.h",
+        "platform/**/*.h",
+    ]),
+    data = [
+        "//tensorflow/core:cuda",
+        "//third_party/gpus/cuda:cublas",
+        "//third_party/gpus/cuda:cudnn",
+    ],
+    linkopts = [
+        "-ldl",
+    ],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//tensorflow/core:lib",
+        "//third_party/gpus/cuda:cuda_headers",
+    ],
+    alwayslink = 1,
+)