aboutsummaryrefslogtreecommitdiffhomepage
path: root/third_party/mkl_dnn/mkldnn.BUILD
diff options
context:
space:
mode:
Diffstat (limited to 'third_party/mkl_dnn/mkldnn.BUILD')
-rw-r--r--third_party/mkl_dnn/mkldnn.BUILD13
1 files changed, 12 insertions, 1 deletions
diff --git a/third_party/mkl_dnn/mkldnn.BUILD b/third_party/mkl_dnn/mkldnn.BUILD
index 58bb7a6a5d..752a0d8498 100644
--- a/third_party/mkl_dnn/mkldnn.BUILD
+++ b/third_party/mkl_dnn/mkldnn.BUILD
@@ -1,5 +1,13 @@
exports_files(["LICENSE"])
+config_setting(
+ name = "clang_linux_x86_64",
+ values = {
+ "cpu": "k8",
+ "define": "using_cuda_clang=true",
+ },
+)
+
cc_library(
name = "mkl_dnn",
srcs = glob([
@@ -9,8 +17,11 @@ cc_library(
hdrs = glob(["include/*"]),
copts = ["-fexceptions"] + select({
"@org_tensorflow//tensorflow:linux_x86_64": [
- "-fopenmp",
+ "-fopenmp", # only works with gcc
],
+ # TODO(ibiryukov): enable openmp with clang by including libomp as a
+ # dependency.
+ ":clang_linux_x86_64": [],
"//conditions:default": [],
}),
includes = [