You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
124 lines
2.6 KiB
124 lines
2.6 KiB
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",
|
|
],
|
|
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",
|
|
],
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
cc_library(
|
|
name = "mkl_libs_windows",
|
|
srcs = [
|
|
"@llvm_openmp//:libiomp5md.dll",
|
|
],
|
|
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"],
|
|
)
|