aboutsummaryrefslogtreecommitdiffhomepage
path: root/third_party/mpi_collectives/BUILD
blob: d5f9edb0b9912bc43621a41f3ca7e090506185d9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
package(default_visibility = ["//visibility:public"])

licenses(["notice"])  # Apache 2.0

exports_files(["LICENSE.txt"])

filegroup(
    name = "all_files",
    srcs = glob(
        ["**/*"],
        exclude = [
            "**/METADATA",
            "**/OWNERS",
        ],
    ),
    visibility = ["//tensorflow:__subpackages__"],
)

cc_library(
    name = "mpi",
    srcs = select({
        "//tensorflow:darwin": ["libmpi.dylib"],
        "//conditions:default": ["libmpi.so"],
    }),
    hdrs = [
        "mpi.h",
        "mpi_portable_platform",
    ],
)