diff options
| -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 |