diff options
author | 2017-05-12 10:05:33 -0700 | |
---|---|---|
committer | 2017-05-12 10:05:33 -0700 | |
commit | b78b5414b8ef61d41bb2bd65bd4618c3555dbf15 (patch) | |
tree | 94ef55c765bd033d7b2d4e58da228dea31c43006 /third_party/mpi/BUILD | |
parent | 45fc73dc09d6316423be98d07e4824870a653d97 (diff) |
MPI based communication path for tensor exchange operations in
distributed TensorFlow
Diffstat (limited to 'third_party/mpi/BUILD')
-rw-r--r-- | third_party/mpi/BUILD | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/third_party/mpi/BUILD b/third_party/mpi/BUILD new file mode 100644 index 0000000000..5f722905b3 --- /dev/null +++ b/third_party/mpi/BUILD @@ -0,0 +1,24 @@ +licenses(["notice"]) # Apache 2.0 + +filegroup( + name = "all_files", + srcs = glob( + ["**/*"], + exclude = [ + "**/METADATA", + "**/OWNERS", + ], + ), + visibility = ["//tensorflow:__subpackages__"], +) + +load("//third_party/mpi:mpi.bzl", "mpi_hdr") + +cc_library( + name = "mpi", + srcs = select({ + "//conditions:default": ["libmpi.so"], + }), + hdrs = mpi_hdr(), + visibility = ["//visibility:public"], +) |