diff options
author | 2018-06-28 23:26:20 +0000 | |
---|---|---|
committer | 2018-06-28 23:26:20 +0000 | |
commit | 83aa3f08966226ac74df5eafb76abb47190b6573 (patch) | |
tree | cfa7c79f9d2d2b7c6bb9700e45f77b7a1c6f641c /tensorflow/stream_executor | |
parent | e83be1d61858aa130809d8b4e3f628c1eb34292d (diff) |
Add stream_executor_headers_lib
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
Diffstat (limited to 'tensorflow/stream_executor')
-rw-r--r-- | tensorflow/stream_executor/BUILD | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/BUILD b/tensorflow/stream_executor/BUILD index 21295abed1..e742f8e8d5 100644 --- a/tensorflow/stream_executor/BUILD +++ b/tensorflow/stream_executor/BUILD @@ -2,6 +2,7 @@ licenses(["restricted"]) load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda_is_configured") load("//tensorflow/core:platform/default/build_config_root.bzl", "if_static") +load("//tensorflow:tensorflow.bzl", "cc_header_only_library") STREAM_EXECUTOR_HEADERS = glob([ "*.h", @@ -51,6 +52,14 @@ cc_library( ] + if_static([":stream_executor_impl"]), ) +cc_header_only_library( + name = "stream_executor_headers_lib", + visibility = ["//visibility:public"], + deps = [ + ":stream_executor", + ], +) + cc_library( name = "cuda_platform", srcs = if_cuda_is_configured( |