diff options
Diffstat (limited to 'tensorflow/core/kernels/BUILD')
| -rw-r--r-- | tensorflow/core/kernels/BUILD | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD index 8940b8cbe4..9f0e7069f5 100644 --- a/tensorflow/core/kernels/BUILD +++ b/tensorflow/core/kernels/BUILD @@ -38,6 +38,7 @@ load( "tf_mkl_kernel_library", "cc_header_only_library", "if_not_windows", + "if_override_eigen_strong_inline", ) load("@local_config_sycl//sycl:build_defs.bzl", "if_sycl") load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test") @@ -3069,6 +3070,10 @@ tf_kernel_library( ":xsmm": ["xsmm_conv2d.h"], "//conditions:default": [], }), + # Override EIGEN_STRONG_INLINE to inline when --define=override_eigen_strong_inline=true, + # So that it doesn't take 20 minutes to compile conv_grad_ops_3d.cc and conv_ops_3d.cc + # on Windows. See https://github.com/tensorflow/tensorflow/issues/10521 + copts = if_override_eigen_strong_inline(["/DEIGEN_STRONG_INLINE=inline"]), defines = select({ ":xsmm": [ "TENSORFLOW_USE_LIBXSMM", |