127 lines
2.8 KiB
Python
127 lines
2.8 KiB
Python
licenses(["notice"]) # 3-Clause BSD
|
|
|
|
load("@bazel_skylib//:bzl_library.bzl", "bzl_library")
|
|
|
|
config_setting(
|
|
name = "build_with_mkl",
|
|
define_values = {
|
|
"build_with_mkl": "true",
|
|
},
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
config_setting(
|
|
name = "build_with_mkl_lnx_x64",
|
|
define_values = {
|
|
"build_with_mkl": "true",
|
|
},
|
|
values = {
|
|
"cpu": "k8",
|
|
},
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
config_setting(
|
|
name = "build_with_mkl_lnx_openmp",
|
|
constraint_values = [
|
|
"@platforms//os:linux",
|
|
],
|
|
define_values = {
|
|
"build_with_mkl": "true",
|
|
"build_with_openmp": "true",
|
|
},
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
config_setting(
|
|
name = "build_with_mkl_windows_openmp",
|
|
constraint_values = [
|
|
"@platforms//os:windows",
|
|
],
|
|
define_values = {
|
|
"build_with_mkl": "true",
|
|
"build_with_openmp": "true",
|
|
},
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
config_setting(
|
|
name = "build_with_mkl_aarch64",
|
|
define_values = {
|
|
"build_with_mkl_aarch64": "true",
|
|
},
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
config_setting(
|
|
name = "enable_mkl",
|
|
define_values = {
|
|
"enable_mkl": "true",
|
|
"build_with_mkl": "true",
|
|
},
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
filegroup(
|
|
name = "LICENSE",
|
|
srcs = [
|
|
"MKL_LICENSE",
|
|
"@llvm_openmp//:LICENSE.txt",
|
|
],
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
# TODO(Intel-tf) Remove the following 3 calls to cc_library and replace all uses
|
|
# of mkl_libs_* with @llvm_openmp//:libiomp5.* directly.
|
|
|
|
cc_library(
|
|
name = "mkl_libs_linux",
|
|
srcs = [
|
|
"@llvm_openmp//:libiomp5.so",
|
|
],
|
|
hdrs = ["@llvm_openmp//:config_omp"],
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
# MacOS build configuration is provided for completness, it has not been tested
|
|
cc_library(
|
|
name = "mkl_libs_darwin",
|
|
srcs = [
|
|
"@llvm_openmp//:libiomp5.dylib",
|
|
],
|
|
hdrs = ["@llvm_openmp//:config_omp"],
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
cc_library(
|
|
name = "mkl_libs_windows",
|
|
srcs = [
|
|
"@llvm_openmp//:libiomp5md.dll",
|
|
],
|
|
hdrs = ["@llvm_openmp//:config_omp"],
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
cc_library(
|
|
name = "intel_binary_blob",
|
|
visibility = ["//visibility:public"],
|
|
deps = select({
|
|
"@org_tensorflow//tensorflow:linux_x86_64": [
|
|
":mkl_libs_linux",
|
|
],
|
|
"@org_tensorflow//tensorflow:macos": [
|
|
":mkl_libs_darwin",
|
|
],
|
|
"@org_tensorflow//tensorflow:windows": [
|
|
":mkl_libs_windows",
|
|
],
|
|
"//conditions:default": [],
|
|
}),
|
|
)
|
|
|
|
bzl_library(
|
|
name = "build_defs_bzl",
|
|
srcs = ["build_defs.bzl"],
|
|
visibility = ["//visibility:public"],
|
|
)
|