diff options
Diffstat (limited to 'tensorflow/tensorflow.bzl')
-rw-r--r-- | tensorflow/tensorflow.bzl | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl index 42753c170b..6cdd02bdca 100644 --- a/tensorflow/tensorflow.bzl +++ b/tensorflow/tensorflow.bzl @@ -138,6 +138,9 @@ def tf_copts(): "//tensorflow:windows": [ "/DLANG_CXX11", "/D__VERSION__=\\\"MSVC\\\"", + "/DPLATFORM_WINDOWS", + "/DEIGEN_HAS_C99_MATH", + "/DTENSORFLOW_USE_EIGEN_THREADPOOL", ], "//tensorflow:ios": ["-std=c++11"], "//conditions:default": ["-pthread"]})) @@ -418,7 +421,7 @@ def _cuda_copts(): # libraries needed by GPU kernels. def tf_gpu_kernel_library(srcs, copts=[], cuda_copts=[], deps=[], hdrs=[], **kwargs): - copts = copts + _cuda_copts() + if_cuda(cuda_copts) + copts = copts + _cuda_copts() + if_cuda(cuda_copts) + tf_copts() native.cc_library( srcs = srcs, |