diff options
Diffstat (limited to 'tensorflow/tensorflow.bzl')
| -rw-r--r-- | tensorflow/tensorflow.bzl | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl index 31a2ee65f9..33c4ff5c3d 100644 --- a/tensorflow/tensorflow.bzl +++ b/tensorflow/tensorflow.bzl @@ -110,8 +110,9 @@ def tf_copts(): "-Wno-sign-compare", "-fno-exceptions",] + if_cuda(["-DGOOGLE_CUDA=1"]) + + if_mkl(["-DINTEL_MKL=1"]) + if_android_arm(["-mfpu=neon"]) + - if_x86(["-msse4.1"]) + + if_x86(["-msse3"]) + select({ "//tensorflow:android": [ "-std=c++11", @@ -481,7 +482,7 @@ def tf_cuda_library(deps=None, cuda_deps=None, copts=None, **kwargs): "//tensorflow/core:cuda", "@local_config_cuda//cuda:cuda_headers" ]), - copts = copts + if_cuda(["-DGOOGLE_CUDA=1"]), + copts = copts + if_cuda(["-DGOOGLE_CUDA=1"]) + if_mkl(["-DINTEL_MKL=1"]), **kwargs) def tf_kernel_library(name, prefix=None, srcs=None, gpu_srcs=None, hdrs=None, |