pytorch

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

3
cc_library(
4
    name = "mkl",
5
    srcs = [
6
        "libmkl_avx2.so",
7
        "libmkl_core.so",
8
        "libmkl_def.so",
9
        "libmkl_intel_lp64.so",
10
        "libmkl_rt.so",
11
        "libmkl_sequential.so",
12
        "libmkl_vml_avx2.so",
13
        "libmkl_vml_avx512.so",
14
        "libmkl_vml_def.so",
15
    ] + select({
16
        "@pytorch//tools/config:thread_sanitizer": [],
17
        "//conditions:default": ["libmkl_tbb_thread.so"],
18
    }),
19
    visibility = ["//visibility:public"],
20
    deps = ["@mkl_headers"],
21
)
22