aboutsummaryrefslogtreecommitdiffhomepage
path: root/configure
diff options
context:
space:
mode:
authorGravatar Jonathan Hseu <jhseu@google.com>2017-07-19 15:04:52 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-07-19 15:12:25 -0700
commit9cc871e81c04ed11829c3364546b4500742140eb (patch)
tree15c55ff29356bddd5a4f252630a9dffc8ac43c4c /configure
parent1fefe92f47ff4149428409847592ec443f1279dd (diff)
Merge changes from github.
END_PUBLIC --- Commit daa67ad17 authored by Jonathan Hseu<vomjom@vomjom.net> Committed by Frank Chen<frankchn@gmail.com>: Remove unittest import (#11596) --- Commit 491beb74c authored by A. Unique TensorFlower<gardener@tensorflow.org> Committed by TensorFlower Gardener<gardener@tensorflow.org>: BEGIN_PUBLIC Automated g4 rollback of changelist 162423171 PiperOrigin-RevId: 162541442
Diffstat (limited to 'configure')
-rwxr-xr-xconfigure20
1 files changed, 8 insertions, 12 deletions
diff --git a/configure b/configure
index 4c6cba2169..93280dc5f5 100755
--- a/configure
+++ b/configure
@@ -26,7 +26,7 @@ function is_windows() {
}
function is_ppc64le() {
- [[ "${uname -m}" == "ppc64le" ]]
+ [[ "$(uname -m)" == "ppc64le" ]]
}
function sed_in_place() {
@@ -298,7 +298,7 @@ fi # TF_NEED_MKL
## Set up architecture-dependent optimization flags.
if [ -z "$CC_OPT_FLAGS" ]; then
- if [ is_ppc64le ]; then
+ if is_ppc64le; then
# gcc on ppc64le does not support -march, use mcpu instead
default_cc_opt_flags="-mcpu=native"
else
@@ -492,6 +492,8 @@ while true; do
if [ -z "$TF_CUDA_VERSION" ]; then
read -p "Please specify the CUDA SDK version you want to use, e.g. 7.0. [Leave empty to default to CUDA 8.0]: " TF_CUDA_VERSION
fi
+ # Set default CUDA version if not set
+ TF_CUDA_VERSION=${TF_CUDA_VERSION:-8.0}
fromuser=""
if [ -z "$CUDA_TOOLKIT_PATH" ]; then
@@ -545,11 +547,7 @@ while true; do
CUDA_TOOLKIT_PATH=""
done
-# Set default CUDA version if not set
-if [ -z "$TF_CUDA_VERSION" ]; then
- TF_CUDA_VERSION="8.0"
- export TF_CUDA_VERSION
-fi
+export TF_CUDA_VERSION
write_action_env_to_bazelrc "TF_CUDA_VERSION" "$TF_CUDA_VERSION"
# Set up which gcc nvcc should use as the host compiler
@@ -587,6 +585,8 @@ while true; do
if [ -z "$TF_CUDNN_VERSION" ]; then
read -p "Please specify the cuDNN version you want to use. [Leave empty to default to cuDNN 6.0]: " TF_CUDNN_VERSION
fi
+ # Set default CUDNN version if not set
+ TF_CUDNN_VERSION=${TF_CUDNN_VERSION:-6}
fromuser=""
if [ -z "$CUDNN_INSTALL_PATH" ]; then
@@ -659,11 +659,7 @@ while true; do
CUDNN_INSTALL_PATH=""
done
-# Set default CUDNN version if not set
-if [ -z "$TF_CUDNN_VERSION" ]; then
- TF_CUDNN_VERSION="6"
- export TF_CUDNN_VERSION
-fi
+export TF_CUDNN_VERSION
write_action_env_to_bazelrc "TF_CUDNN_VERSION" "$TF_CUDNN_VERSION"
# Configure the compute capabilities that TensorFlow builds for.