From 1e67c90e2caceeff82d09793d1ef5fa0300d219b Mon Sep 17 00:00:00 2001 From: Peter Hawkins Date: Mon, 9 Jan 2017 12:04:37 -0800 Subject: Initial open-source release of XLA: Accelerated Linear Algebra. XLA is a compiler-based linear algebra execution engine that targets CPUs, GPUs and custom accelerators. XLA is still experimental; we are releasing it early to get the community involved. Change: 143990941 --- configure | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) (limited to 'configure') diff --git a/configure b/configure index 8d4b12aad2..64add33bd5 100755 --- a/configure +++ b/configure @@ -112,6 +112,26 @@ else sed -i -e "s/WITH_HDFS_SUPPORT = True/WITH_HDFS_SUPPORT = False/" tensorflow/core/platform/default/build_config.bzl fi +## Enable XLA. +while [ "$TF_ENABLE_XLA" == "" ]; do + read -p "Do you wish to build TensorFlow with the XLA just-in-time compiler (experimental)? [y/N] " INPUT + case $INPUT in + [Yy]* ) echo "XLA JIT support will be enabled for TensorFlow"; TF_ENABLE_XLA=1;; + [Nn]* ) echo "No XLA JIT support will be enabled for TensorFlow"; TF_ENABLE_XLA=0;; + "" ) echo "No XLA support will be enabled for TensorFlow"; TF_ENABLE_XLA=0;; + * ) echo "Invalid selection: " $INPUT;; + esac +done + +if [ "$TF_ENABLE_XLA" == "1" ]; then + # Update Bazel build configuration. + perl -pi -e "s,WITH_XLA_SUPPORT = (False|True),WITH_XLA_SUPPORT = True,s" tensorflow/core/platform/default/build_config.bzl +else + # Update Bazel build configuration. + perl -pi -e "s,WITH_XLA_SUPPORT = (False|True),WITH_XLA_SUPPORT = False,s" tensorflow/core/platform/default/build_config.bzl +fi + + # Invoke python_config and set up symlinks to python includes ./util/python/python_config.sh --setup "$PYTHON_BIN_PATH" -- cgit v1.2.3