xref: /aosp_15_r20/external/pytorch/third_party/ideep.BUILD (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1*da0073e9SAndroid Build Coastguard Workerload("@rules_cc//cc:defs.bzl", "cc_library")
2*da0073e9SAndroid Build Coastguard Worker
3*da0073e9SAndroid Build Coastguard Workercc_library(
4*da0073e9SAndroid Build Coastguard Worker    name = "ideep",
5*da0073e9SAndroid Build Coastguard Worker    hdrs = glob([
6*da0073e9SAndroid Build Coastguard Worker        "include/**/*.hpp",
7*da0073e9SAndroid Build Coastguard Worker        "include/**/*.h",
8*da0073e9SAndroid Build Coastguard Worker    ]),
9*da0073e9SAndroid Build Coastguard Worker    defines = [
10*da0073e9SAndroid Build Coastguard Worker        "IDEEP_USE_MKL",
11*da0073e9SAndroid Build Coastguard Worker    ],
12*da0073e9SAndroid Build Coastguard Worker    includes = [
13*da0073e9SAndroid Build Coastguard Worker        "include/",
14*da0073e9SAndroid Build Coastguard Worker    ],
15*da0073e9SAndroid Build Coastguard Worker    visibility = ["//visibility:public"],
16*da0073e9SAndroid Build Coastguard Worker    deps = ["@mkl_dnn//:mkl-dnn"],
17*da0073e9SAndroid Build Coastguard Worker)
18