Build TensorFlow with oneDNN primitives only for `linux_86_64` and `windows`.
PiperOrigin-RevId: 364857561
Change-Id: I5b46519229b714f50a20511e76210f46257db05d
diff --git a/third_party/mkl/build_defs.bzl b/third_party/mkl/build_defs.bzl
index 4643b11..d913f22 100644
--- a/third_party/mkl/build_defs.bzl
+++ b/third_party/mkl/build_defs.bzl
@@ -33,14 +33,9 @@
"""
return select({
"@org_tensorflow//third_party/mkl:build_with_mkl": if_true,
- "//tensorflow:android_x86": if_false,
- "//tensorflow:android_x86_64": if_false,
- "//tensorflow:arm_any": if_false,
- "//tensorflow:ios": if_false,
- "//tensorflow:linux_ppc64le": if_false,
- "//tensorflow:linux_s390x": if_false,
- "//tensorflow:macos": if_false,
- "//conditions:default": if_true,
+ "//tensorflow:linux_x86_64": if_true,
+ "//tensorflow:windows": if_true,
+ "//conditions:default": if_false,
})
def if_mkl_ml(if_true, if_false = []):
@@ -106,15 +101,11 @@
inclusion in the deps attribute of rules.
"""
return select({
+ "@org_tensorflow//third_party/mkl:build_with_mkl": ["@mkl_dnn_v1//:mkl_dnn"],
"@org_tensorflow//third_party/mkl:build_with_mkl_aarch64": ["@mkl_dnn_v1//:mkl_dnn_aarch64"],
- "//tensorflow:android_x86": [],
- "//tensorflow:android_x86_64": [],
- "//tensorflow:arm_any": [],
- "//tensorflow:ios": [],
- "//tensorflow:linux_ppc64le": [],
- "//tensorflow:linux_s390x": [],
- "//tensorflow:macos": [],
- "//conditions:default": ["@mkl_dnn_v1//:mkl_dnn"],
+ "//tensorflow:linux_x86_64": ["@mkl_dnn_v1//:mkl_dnn"],
+ "//tensorflow:windows": ["@mkl_dnn_v1//:mkl_dnn"],
+ "//conditions:default": [],
})
def _enable_local_mkl(repository_ctx):
