aboutsummaryrefslogtreecommitdiffhomepage
path: root/third_party/mkl_dnn/mkldnn.BUILD
blob: 58bb7a6a5d0494301aa5b0bd29f858e7d06e69d3 (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
exports_files(["LICENSE"])

cc_library(
    name = "mkl_dnn",
    srcs = glob([
        "src/common/*.cpp",
        "src/cpu/*.cpp",
    ]),
    hdrs = glob(["include/*"]),
    copts = ["-fexceptions"] + select({
        "@org_tensorflow//tensorflow:linux_x86_64": [
            "-fopenmp",
        ],
        "//conditions:default": [],
    }),
    includes = [
        "include",
        "src",
        "src/common",
        "src/cpu",
        "src/cpu/xbyak",
    ],
    nocopts = "-fno-exceptions",
    visibility = ["//visibility:public"],
)