diff options
author | 2018-07-19 17:22:25 -0700 | |
---|---|---|
committer | 2018-07-19 17:25:04 -0700 | |
commit | 78121905a545ca8e91ab1dbc899b5509933331d7 (patch) | |
tree | 2bdc04a130bec8c92df1c917fdff16044920ebca /tensorflow | |
parent | e2754fe9098299c11e2c3214513a1fe4732b412f (diff) |
Set correct TF_OVERRIDE_EIGEN_STRONG_INLINE value in build_tf_windows.sh
PiperOrigin-RevId: 205329361
Diffstat (limited to 'tensorflow')
-rw-r--r-- | tensorflow/tools/ci_build/windows/cpu/pip/build_tf_windows.sh | 4 | ||||
-rw-r--r-- | tensorflow/tools/ci_build/windows/gpu/pip/build_tf_windows.sh | 4 |
2 files changed, 6 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 dc7ea1dc57..42f58deb42 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 @@ -66,11 +66,13 @@ for ARG in "$@"; do fi done -if [[ "$release_build" != 1 ]]; then +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 +else + export TF_OVERRIDE_EIGEN_STRONG_INLINE=1 fi # Enable short object file path to avoid long path issue on Windows. 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 a4175a0e81..2a8c2d9167 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 @@ -66,11 +66,13 @@ for ARG in "$@"; do fi done -if [[ "$release_build" != 1 ]]; then +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 +else + export TF_OVERRIDE_EIGEN_STRONG_INLINE=1 fi # Enable short object file path to avoid long path issue on Windows. |