pytorch

1
load("@rules_cc//cc:defs.bzl", "cc_library")
2

3
cc_library(
4
    name = "ideep",
5
    hdrs = glob([
6
        "include/**/*.hpp",
7
        "include/**/*.h",
8
    ]),
9
    defines = [
10
        "IDEEP_USE_MKL",
11
    ],
12
    includes = [
13
        "include/",
14
    ],
15
    visibility = ["//visibility:public"],
16
    deps = ["@mkl_dnn//:mkl-dnn"],
17
)
18