xref: /aosp_15_r20/external/pytorch/third_party/ideep.BUILD (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1load("@rules_cc//cc:defs.bzl", "cc_library")
2
3cc_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