fe1bce6717
PR #42339: [INTEL MKL] MKL DNN v0.x cleanup - Reset MKL build config and remove DNN v0.x related macros Imported from GitHub PR https://github.com/tensorflow/tensorflow/pull/42339 This is the first PR for MKL DNN v0.x clean - for Tensorflow 2.4, only MKL DNN v1.x will be supported (1) Reset MKL DNN related build config (mostly in third_part/mkl or mkl_dnn folder) (2) Remove MKL DNN v0.x related macros in core/util/mkl_types.c (3) Minor code style fix in one MKL kernel op. There will be a sequence of PR's which will clean up all related MKL kernels ops. PiperOrigin-RevId: 331205640 Change-Id: I2c3c3671fe8906a5dd18c6b017d05fc69a1e3f59
35 lines
741 B
Python
35 lines
741 B
Python
load("@bazel_skylib//:bzl_library.bzl", "bzl_library")
|
|
|
|
package(
|
|
default_visibility = [
|
|
"//tensorflow:__subpackages__",
|
|
],
|
|
licenses = ["notice"],
|
|
)
|
|
|
|
exports_files(["LICENSE"])
|
|
|
|
config_setting(
|
|
name = "build_with_mkl_opensource",
|
|
define_values = {
|
|
"build_with_mkl": "true",
|
|
"build_with_mkl_opensource": "true",
|
|
},
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
config_setting(
|
|
name = "build_with_mkldnn_threadpool",
|
|
define_values = {
|
|
"build_with_mkl": "true",
|
|
"build_with_mkl_opensource": "true",
|
|
"build_with_mkldnn_threadpool": "true",
|
|
},
|
|
visibility = ["//visibility:public"],
|
|
)
|
|
|
|
bzl_library(
|
|
name = "build_defs_bzl",
|
|
srcs = ["build_defs.bzl"],
|
|
)
|