diff options
| author |  Qianli Scott Zhu <scottzhu@google.com> | 2018-06-29 13:13:51 -0700 |
|---|---|---|
| committer |  GitHub <noreply@github.com> | 2018-06-29 13:13:51 -0700 |
| commit | 5ac1bd7836e0f1a4d3b02f9538c4277914c8e5f5 (patch) | |
| tree | bb99677bd4c3495618d1c280d883d86b0c75725c /tensorflow/stream_executor | |
| parent | 01c36c3d7b3e230c865e71d67e138a8dc765e7a6 (diff) | |
| parent | 83aa3f08966226ac74df5eafb76abb47190b6573 (diff) | |
Merge pull request #20147 from yongtang/06202018-mpi_collectives
Fix MPI build issue with bazel
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( |