From b84b607c4b2ba68d61e17bc4f404bd6d8aa6ec9a Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Fri, 22 Nov 2019 15:02:00 -0800 Subject: [PATCH] Override EIGEN strong inline for release builds as well. PiperOrigin-RevId: 282049163 Change-Id: I2fefa6ed9198aee013ce3bdc07d21dfb127833a7 --- tensorflow/tools/ci_build/windows/cpu/pip/build_tf_windows.sh | 2 +- tensorflow/tools/ci_build/windows/gpu/pip/build_tf_windows.sh | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/tools/ci_build/windows/cpu/pip/build_tf_windows.sh b/tensorflow/tools/ci_build/windows/cpu/pip/build_tf_windows.sh index faddfbf1462..c21fbd80bb7 100644 --- a/tensorflow/tools/ci_build/windows/cpu/pip/build_tf_windows.sh +++ b/tensorflow/tools/ci_build/windows/cpu/pip/build_tf_windows.sh @@ -104,7 +104,7 @@ if [[ "$RELEASE_BUILD" == 1 ]]; then # Overriding eigen strong inline speeds up the compiling of conv_grad_ops_3d.cc and conv_ops_3d.cc # by 20 minutes. See https://github.com/tensorflow/tensorflow/issues/10521 # Because this hurts the performance of TF, we don't override it in release build. - export TF_OVERRIDE_EIGEN_STRONG_INLINE=0 + export TF_OVERRIDE_EIGEN_STRONG_INLINE=1 else export TF_OVERRIDE_EIGEN_STRONG_INLINE=1 fi diff --git a/tensorflow/tools/ci_build/windows/gpu/pip/build_tf_windows.sh b/tensorflow/tools/ci_build/windows/gpu/pip/build_tf_windows.sh index 45bde842060..da2f5d275c7 100644 --- a/tensorflow/tools/ci_build/windows/gpu/pip/build_tf_windows.sh +++ b/tensorflow/tools/ci_build/windows/gpu/pip/build_tf_windows.sh @@ -104,7 +104,7 @@ if [[ "$RELEASE_BUILD" == 1 ]]; then # Overriding eigen strong inline speeds up the compiling of conv_grad_ops_3d.cc and conv_ops_3d.cc # by 20 minutes. See https://github.com/tensorflow/tensorflow/issues/10521 # Because this hurts the performance of TF, we don't override it in release build. - export TF_OVERRIDE_EIGEN_STRONG_INLINE=0 + export TF_OVERRIDE_EIGEN_STRONG_INLINE=1 else export TF_OVERRIDE_EIGEN_STRONG_INLINE=1 fi