diff options
Diffstat (limited to 'tensorflow/tensorflow.bzl')
-rw-r--r-- | tensorflow/tensorflow.bzl | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl index 690af5485a..5b1d9a517e 100644 --- a/tensorflow/tensorflow.bzl +++ b/tensorflow/tensorflow.bzl @@ -147,7 +147,7 @@ def tf_copts(): "-Iexternal/gemmlowp", "-Wno-sign-compare", "-fno-exceptions", - ] + if_cuda(["-DGOOGLE_CUDA=1"]) + if_mkl(["-DINTEL_MKL=1"]) + if_android_arm( + ] + if_cuda(["-DGOOGLE_CUDA=1"]) + if_mkl(["-DINTEL_MKL=1", "-fopenmp",]) + if_android_arm( ["-mfpu=neon"]) + if_x86(["-msse3"]) + select({ clean_dep("//tensorflow:android"): [ "-std=c++11", |