experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Override EIGEN strong inline for release builds as well.

Browse Source 
PiperOrigin-RevId: 282049163
Change-Id: I2fefa6ed9198aee013ce3bdc07d21dfb127833a7
This commit is contained in:
A. Unique TensorFlower 2019-11-22 15:02:00 -08:00 committed by TensorFlower Gardener
parent 4235b604cf
commit b84b607c4b
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/tools/ci_build/windows/cpu/pip/build_tf_windows.sh
View File

@ -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

2
tensorflow/tools/ci_build/windows/gpu/pip/build_tf_windows.sh
View File

@ -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