diff options
| author |  Todd Wang <toddw@google.com> | 2018-09-28 08:56:06 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-28 09:00:05 -0700 |
| commit | 35459cbaa0f654393b242c5357f6939b05267ab8 (patch) | |
| tree | 4a17adf8d8826a1a54e8dc32e4aca21ae0840fb3 | |
| parent | 9ef0ec921cc6de670fd2fdba1be49e0eca2a1043 (diff) | |
Build TF with XLA support by default.
Building binaries with XLA support does not enable it by default, it
simply makes it accessible via default binary builds.
PiperOrigin-RevId: 214942824
| -rw-r--r-- | configure.py | 4 | ||||
| -rwxr-xr-x | tensorflow/tools/ci_build/builds/run_pip_tests.sh | 1 |
2 files changed, 2 insertions, 3 deletions
diff --git a/configure.py b/configure.py index 55fce8b93b..9899ae10e8 100644 --- a/configure.py +++ b/configure.py @@ -1504,6 +1504,7 @@ def main(): if is_macos(): environ_cp['TF_NEED_JEMALLOC'] = '0' environ_cp['TF_NEED_TENSORRT'] = '0' + environ_cp['TF_ENABLE_XLA'] = '0' # The numpy package on ppc64le uses OpenBLAS which has multi-threading # issues that lead to incorrect answers. Set OMP_NUM_THREADS=1 at @@ -1515,7 +1516,7 @@ def main(): set_build_var(environ_cp, 'TF_NEED_JEMALLOC', 'jemalloc as malloc', 'with_jemalloc', True) set_build_var(environ_cp, 'TF_ENABLE_XLA', 'XLA JIT', 'with_xla_support', - False, 'xla') + True, 'xla') set_action_env_var(environ_cp, 'TF_NEED_OPENCL_SYCL', 'OpenCL SYCL', False) @@ -1624,4 +1625,3 @@ def main(): if __name__ == '__main__': main() - diff --git a/tensorflow/tools/ci_build/builds/run_pip_tests.sh b/tensorflow/tools/ci_build/builds/run_pip_tests.sh index 17198a6560..7d5cf3f843 100755 --- a/tensorflow/tools/ci_build/builds/run_pip_tests.sh +++ b/tensorflow/tools/ci_build/builds/run_pip_tests.sh @@ -111,7 +111,6 @@ bazel clean # virtualenv. export TF_NEED_GCP=0 export TF_NEED_HDFS=0 -export TF_ENABLE_XLA=0 # Obtain the path to Python binary if [[ ${IS_VIRTUALENV} == "1" ]]; then |