diff options
Diffstat (limited to 'tensorflow/tensorflow.bzl')
-rw-r--r-- | tensorflow/tensorflow.bzl | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl index 83a9313e50..cddf16f147 100644 --- a/tensorflow/tensorflow.bzl +++ b/tensorflow/tensorflow.bzl @@ -117,11 +117,9 @@ def if_not_windows(a): }) -def if_x86(a): +def if_linux_x86_64(a): return select({ clean_dep("//tensorflow:linux_x86_64"): a, - clean_dep("//tensorflow:windows"): a, - clean_dep("//tensorflow:windows_msvc"): a, "//conditions:default": [], }) @@ -152,7 +150,7 @@ def tf_copts(): "-Wno-sign-compare", "-fno-exceptions", ]) + if_cuda(["-DGOOGLE_CUDA=1"]) + if_mkl(["-DINTEL_MKL=1", "-fopenmp",]) + if_android_arm( - ["-mfpu=neon"]) + if_x86(["-msse3"]) + select({ + ["-mfpu=neon"]) + if_linux_x86_64(["-msse3"]) + select({ clean_dep("//tensorflow:android"): [ "-std=c++11", "-DTF_LEAN_BINARY", |