xref: /aosp_15_r20/external/tensorflow/third_party/mkl/BUILD (revision b6fb3261f9314811a0f4371741dbb8839866f948) 
1licenses(["notice"])  # 3-Clause BSD
2
3load("@bazel_skylib//:bzl_library.bzl", "bzl_library")
4
5config_setting(
6    name = "build_with_mkl",
7    define_values = {
8        "build_with_mkl": "true",
9    },
10    visibility = ["//visibility:public"],
11)
12
13config_setting(
14    name = "build_with_mkl_lnx_x64",
15    define_values = {
16        "build_with_mkl": "true",
17    },
18    values = {
19        "cpu": "k8",
20    },
21    visibility = ["//visibility:public"],
22)
23
24config_setting(
25    name = "build_with_mkl_lnx_openmp",
26    constraint_values = [
27        "@platforms//os:linux",
28    ],
29    define_values = {
30        "build_with_mkl": "true",
31        "build_with_openmp": "true",
32    },
33    visibility = ["//visibility:public"],
34)
35
36config_setting(
37    name = "build_with_mkl_windows_openmp",
38    constraint_values = [
39        "@platforms//os:windows",
40    ],
41    define_values = {
42        "build_with_mkl": "true",
43        "build_with_openmp": "true",
44    },
45    visibility = ["//visibility:public"],
46)
47
48config_setting(
49    name = "build_with_mkl_aarch64",
50    define_values = {
51        "build_with_mkl_aarch64": "true",
52    },
53    visibility = ["//visibility:public"],
54)
55
56config_setting(
57    name = "enable_mkl",
58    define_values = {
59        "enable_mkl": "true",
60        "build_with_mkl": "true",
61    },
62    visibility = ["//visibility:public"],
63)
64
65filegroup(
66    name = "LICENSE",
67    srcs = [
68        "MKL_LICENSE",
69        "@llvm_openmp//:LICENSE.txt",
70    ],
71    visibility = ["//visibility:public"],
72)
73
74# TODO(Intel-tf) Remove the following 3 calls to cc_library and replace all uses
75# of mkl_libs_* with @llvm_openmp//:libiomp5.* directly.
76
77cc_library(
78    name = "mkl_libs_linux",
79    srcs = [
80        "@llvm_openmp//:libiomp5.so",
81    ],
82    hdrs = ["@llvm_openmp//:config_omp"],
83    visibility = ["//visibility:public"],
84)
85
86# MacOS build configuration is provided for completness, it has not been tested
87cc_library(
88    name = "mkl_libs_darwin",
89    srcs = [
90        "@llvm_openmp//:libiomp5.dylib",
91    ],
92    hdrs = ["@llvm_openmp//:config_omp"],
93    visibility = ["//visibility:public"],
94)
95
96cc_library(
97    name = "mkl_libs_windows",
98    srcs = [
99        "@llvm_openmp//:libiomp5md.dll",
100    ],
101    hdrs = ["@llvm_openmp//:config_omp"],
102    visibility = ["//visibility:public"],
103)
104
105cc_library(
106    name = "intel_binary_blob",
107    visibility = ["//visibility:public"],
108    deps = select({
109        "@org_tensorflow//tensorflow:linux_x86_64": [
110            ":mkl_libs_linux",
111        ],
112        "@org_tensorflow//tensorflow:macos": [
113            ":mkl_libs_darwin",
114        ],
115        "@org_tensorflow//tensorflow:windows": [
116            ":mkl_libs_windows",
117        ],
118        "//conditions:default": [],
119    }),
120)
121
122bzl_library(
123    name = "build_defs_bzl",
124    srcs = ["build_defs.bzl"],
125    visibility = ["//visibility:public"],
126)
127