Geoffrey Irving | 4c85a08 | 2016-03-16 12:20:34 -0800 | [diff] [blame] | 1 | #!/usr/bin/env bash |
Manjunath Kudlur | f41959c | 2015-11-06 16:27:58 -0800 | [diff] [blame] | 2 | |
A. Unique TensorFlower | edaf3b3 | 2016-10-10 10:26:22 -0800 | [diff] [blame] | 3 | set -e |
| 4 | set -o pipefail |
| 5 | |
Andrew Selle | 09045e4 | 2016-09-06 08:19:04 -0800 | [diff] [blame] | 6 | # Find out the absolute path to where ./configure resides |
A. Unique TensorFlower | 46d2c28 | 2017-01-02 22:19:48 -0800 | [diff] [blame] | 7 | pushd `dirname $0` > /dev/null |
Andrew Selle | 09045e4 | 2016-09-06 08:19:04 -0800 | [diff] [blame] | 8 | SOURCE_BASE_DIR=`pwd -P` |
| 9 | popd > /dev/null |
| 10 | |
Patrick Nguyen | c5ab3dd | 2016-10-20 12:09:18 -0800 | [diff] [blame] | 11 | PLATFORM="$(uname -s | tr 'A-Z' 'a-z')" |
Jonathan Hseu | c058a01 | 2017-01-17 15:06:43 -0800 | [diff] [blame] | 12 | |
| 13 | function is_linux() { |
| 14 | if [[ "${PLATFORM}" == "linux" ]]; then |
| 15 | true |
| 16 | else |
| 17 | false |
| 18 | fi |
| 19 | } |
| 20 | |
| 21 | function is_macos() { |
| 22 | if [[ "${PLATFORM}" == "darwin" ]]; then |
| 23 | true |
| 24 | else |
| 25 | false |
| 26 | fi |
| 27 | } |
| 28 | |
Patrick Nguyen | c5ab3dd | 2016-10-20 12:09:18 -0800 | [diff] [blame] | 29 | function is_windows() { |
| 30 | # On windows, the shell script is actually running in msys |
Shanqing Cai | 56fc883 | 2017-01-23 18:25:25 -0800 | [diff] [blame] | 31 | if [[ "${PLATFORM}" =~ msys_nt*|mingw*|cygwin*|uwin* ]]; then |
Patrick Nguyen | c5ab3dd | 2016-10-20 12:09:18 -0800 | [diff] [blame] | 32 | true |
| 33 | else |
| 34 | false |
| 35 | fi |
| 36 | } |
| 37 | |
Dandelion Mané | 0386a01 | 2017-03-10 14:43:23 -0800 | [diff] [blame] | 38 | function sed_hyphen_i() { |
| 39 | if is_macos; then |
| 40 | sed -i '' "$@" |
| 41 | else |
| 42 | sed -i "$@" |
| 43 | fi |
| 44 | } |
| 45 | |
A. Unique TensorFlower | ccbc899 | 2017-04-04 16:10:08 -0800 | [diff] [blame] | 46 | function write_to_bazelrc() { |
| 47 | echo "$1" >> .tf_configure.bazelrc |
| 48 | } |
| 49 | |
| 50 | function write_action_env_to_bazelrc() { |
| 51 | write_to_bazelrc "build --action_env $1=\"$2\"" |
| 52 | } |
| 53 | |
| 54 | # This file contains customized config settings. |
| 55 | rm -f .tf_configure.bazelrc |
| 56 | touch .tf_configure.bazelrc |
| 57 | touch .bazelrc |
| 58 | sed_hyphen_i "/tf_configure/d" .bazelrc |
A. Unique TensorFlower | 8268476 | 2017-04-05 11:30:37 -0800 | [diff] [blame] | 59 | echo "import %workspace%/.tf_configure.bazelrc" >> .bazelrc |
A. Unique TensorFlower | ccbc899 | 2017-04-04 16:10:08 -0800 | [diff] [blame] | 60 | |
Andrew Harp | 51dbc46 | 2017-01-27 14:42:30 -0800 | [diff] [blame] | 61 | # Delete any leftover BUILD files from the Makefile build, which would interfere |
| 62 | # with Bazel parsing. |
| 63 | MAKEFILE_DOWNLOAD_DIR=tensorflow/contrib/makefile/downloads |
| 64 | if [ -d "${MAKEFILE_DOWNLOAD_DIR}" ]; then |
| 65 | find ${MAKEFILE_DOWNLOAD_DIR} -type f -name '*BUILD' -delete |
| 66 | fi |
| 67 | |
Vijay Vasudevan | bf6b536 | 2015-12-02 15:04:40 -0800 | [diff] [blame] | 68 | ## Set up python-related environment settings |
| 69 | while true; do |
| 70 | fromuser="" |
| 71 | if [ -z "$PYTHON_BIN_PATH" ]; then |
A. Unique TensorFlower | edaf3b3 | 2016-10-10 10:26:22 -0800 | [diff] [blame] | 72 | default_python_bin_path=$(which python || which python3 || true) |
Vijay Vasudevan | bf6b536 | 2015-12-02 15:04:40 -0800 | [diff] [blame] | 73 | read -p "Please specify the location of python. [Default is $default_python_bin_path]: " PYTHON_BIN_PATH |
| 74 | fromuser="1" |
| 75 | if [ -z "$PYTHON_BIN_PATH" ]; then |
| 76 | PYTHON_BIN_PATH=$default_python_bin_path |
| 77 | fi |
| 78 | fi |
| 79 | if [ -e "$PYTHON_BIN_PATH" ]; then |
| 80 | break |
| 81 | fi |
| 82 | echo "Invalid python path. ${PYTHON_BIN_PATH} cannot be found" 1>&2 |
| 83 | if [ -z "$fromuser" ]; then |
| 84 | exit 1 |
| 85 | fi |
| 86 | PYTHON_BIN_PATH="" |
| 87 | # Retry |
| 88 | done |
A. Unique TensorFlower | 8e0d6f1 | 2017-04-21 06:35:18 -0800 | [diff] [blame] | 89 | export PYTHON_BIN_PATH |
| 90 | write_action_env_to_bazelrc "PYTHON_BIN_PATH" "$PYTHON_BIN_PATH" |
| 91 | # TODO(ngiraldo): allow the user to optionally set PYTHON_INCLUDE_PATH and NUMPY_INCLUDE_PATH |
Vijay Vasudevan | bf6b536 | 2015-12-02 15:04:40 -0800 | [diff] [blame] | 92 | |
Benoit Steiner | 639b4e7 | 2017-02-08 09:25:09 -0800 | [diff] [blame] | 93 | ## Set up MKL related environment settings |
| 94 | if false; then # Disable building with MKL for now |
| 95 | while [ "$TF_NEED_MKL" == "" ]; do |
| 96 | fromuser="" |
| 97 | read -p "Do you wish to build TensorFlow with MKL support? [y/N] " INPUT |
| 98 | fromuser="1" |
| 99 | case $INPUT in |
| 100 | [Yy]* ) echo "MKL support will be enabled for TensorFlow"; TF_NEED_MKL=1;; |
| 101 | [Nn]* ) echo "No MKL support will be enabled for TensorFlow"; TF_NEED_MKL=0;; |
| 102 | "" ) echo "No MKL support will be enabled for TensorFlow"; TF_NEED_MKL=0;; |
| 103 | * ) echo "Invalid selection: " $INPUT;; |
| 104 | esac |
| 105 | done |
| 106 | |
| 107 | OSNAME=`uname -s` |
| 108 | |
| 109 | if [ "$TF_NEED_MKL" == "1" ]; then # TF_NEED_MKL |
| 110 | DST=`dirname $0` |
Martin Wicke | bc456e3 | 2017-03-23 12:31:16 -0800 | [diff] [blame] | 111 | ARCHIVE_BASENAME=mklml_lnx_2017.0.2.20170209.tgz |
| 112 | GITHUB_RELEASE_TAG=v0.5 |
Benoit Steiner | 639b4e7 | 2017-02-08 09:25:09 -0800 | [diff] [blame] | 113 | MKLURL="https://github.com/01org/mkl-dnn/releases/download/$GITHUB_RELEASE_TAG/$ARCHIVE_BASENAME" |
| 114 | if ! [ -e "$DST/third_party/mkl/$ARCHIVE_BASENAME" ]; then |
| 115 | wget --no-check-certificate -P $DST/third_party/mkl/ $MKLURL |
| 116 | fi |
| 117 | tar -xzf $DST/third_party/mkl/$ARCHIVE_BASENAME -C $DST/third_party/mkl/ |
| 118 | extracted_dir_name="${ARCHIVE_BASENAME%.*}" |
| 119 | MKL_INSTALL_PATH=$DST/third_party/mkl/$extracted_dir_name |
| 120 | MKL_INSTALL_PATH=`${PYTHON_BIN_PATH} -c "import os; print(os.path.realpath(os.path.expanduser('${MKL_INSTALL_PATH}')))"` |
| 121 | |
| 122 | if [ "$OSNAME" == "Linux" ]; then |
| 123 | # Full MKL configuration |
| 124 | MKL_RT_LIB_PATH="lib/intel64/libmkl_rt.so" #${TF_MKL_EXT}#TODO version? |
| 125 | MKL_RT_OMP_LIB_PATH="../compiler/lib/intel64/libiomp5.so" #TODO VERSION? |
| 126 | |
| 127 | # MKL-ML configuration |
| 128 | MKL_ML_LIB_PATH="lib/libmklml_intel.so" #${TF_MKL_EXT}#TODO version? |
| 129 | MKL_ML_OMP_LIB_PATH="lib/libiomp5.so" #TODO VERSION? |
| 130 | elif [ "$OSNAME" == "Darwin" ]; then |
| 131 | echo "Darwin is unsupported yet"; |
| 132 | exit 1 |
| 133 | fi |
| 134 | |
| 135 | if [ -e "$MKL_INSTALL_PATH/${MKL_ML_LIB_PATH}" ]; then |
| 136 | ln -sf $MKL_INSTALL_PATH/${MKL_ML_LIB_PATH} third_party/mkl/ |
| 137 | ln -sf $MKL_INSTALL_PATH/${MKL_ML_OMP_LIB_PATH} third_party/mkl/ |
| 138 | ln -sf $MKL_INSTALL_PATH/include third_party/mkl/ |
| 139 | ln -sf $MKL_INSTALL_PATH/include third_party/eigen3/mkl_include |
| 140 | else |
| 141 | echo "ERROR: $MKL_INSTALL_PATH/${MKL_ML_LIB_PATH} does not exist"; |
| 142 | exit 1 |
| 143 | fi |
| 144 | |
| 145 | if [ -z "$fromuser" ]; then |
| 146 | exit 1 |
| 147 | fi |
| 148 | |
| 149 | cat > third_party/mkl/mkl.config <<EOF |
| 150 | # MKL_INSTALL_PATH refers to the location of MKL root folder. The MKL header and library |
| 151 | # files can be either in this directory, or under include/ and lib64/ |
| 152 | MKL_INSTALL_PATH=$MKL_INSTALL_PATH |
| 153 | EOF |
| 154 | |
| 155 | fi # TF_NEED_MKL |
| 156 | ################## MKL |
| 157 | fi # Disable building with MKL for now |
| 158 | |
Martin Wicke | c4e3d4a | 2017-01-13 12:20:42 -0800 | [diff] [blame] | 159 | ## Set up architecture-dependent optimization flags. |
| 160 | if [ -z "$CC_OPT_FLAGS" ]; then |
| 161 | default_cc_opt_flags="-march=native" |
Benoit Steiner | 639b4e7 | 2017-02-08 09:25:09 -0800 | [diff] [blame] | 162 | read -p "Please specify optimization flags to use during compilation when bazel option "\ |
| 163 | "\"--config=opt\" is specified [Default is $default_cc_opt_flags]: " CC_OPT_FLAGS |
Martin Wicke | c4e3d4a | 2017-01-13 12:20:42 -0800 | [diff] [blame] | 164 | if [ -z "$CC_OPT_FLAGS" ]; then |
| 165 | CC_OPT_FLAGS=$default_cc_opt_flags |
| 166 | fi |
| 167 | fi |
| 168 | |
Patrick Nguyen | c5ab3dd | 2016-10-20 12:09:18 -0800 | [diff] [blame] | 169 | if is_windows; then |
A. Unique TensorFlower | 95a954a | 2016-12-28 13:40:08 -0800 | [diff] [blame] | 170 | TF_NEED_GCP=0 |
Patrick Nguyen | c5ab3dd | 2016-10-20 12:09:18 -0800 | [diff] [blame] | 171 | TF_NEED_HDFS=0 |
Jonathan Hseu | 83c6e0c | 2017-01-11 16:39:35 -0800 | [diff] [blame] | 172 | TF_NEED_JEMALLOC=0 |
Benoit Steiner | a771598 | 2016-11-09 13:14:03 -0800 | [diff] [blame] | 173 | TF_NEED_OPENCL=0 |
A. Unique TensorFlower | ccbc899 | 2017-04-04 16:10:08 -0800 | [diff] [blame] | 174 | TF_CUDA_CLANG=0 |
Patrick Nguyen | c5ab3dd | 2016-10-20 12:09:18 -0800 | [diff] [blame] | 175 | fi |
| 176 | |
Jonathan Hseu | c058a01 | 2017-01-17 15:06:43 -0800 | [diff] [blame] | 177 | if is_linux; then |
| 178 | while [ "$TF_NEED_JEMALLOC" == "" ]; do |
| 179 | read -p "Do you wish to use jemalloc as the malloc implementation? [Y/n] "\ |
| 180 | INPUT |
| 181 | case $INPUT in |
| 182 | [Yy]* ) echo "jemalloc enabled"; TF_NEED_JEMALLOC=1;; |
| 183 | [Nn]* ) echo "jemalloc disabled"; TF_NEED_JEMALLOC=0;; |
| 184 | "" ) echo "jemalloc enabled"; TF_NEED_JEMALLOC=1;; |
| 185 | * ) echo "Invalid selection: " $INPUT;; |
| 186 | esac |
| 187 | done |
| 188 | else |
| 189 | TF_NEED_JEMALLOC=0 |
| 190 | fi |
Jonathan Hseu | 83c6e0c | 2017-01-11 16:39:35 -0800 | [diff] [blame] | 191 | |
Martin Wicke | bc456e3 | 2017-03-23 12:31:16 -0800 | [diff] [blame] | 192 | if [[ "$TF_NEED_JEMALLOC" == "1" ]]; then |
A. Unique TensorFlower | ccbc899 | 2017-04-04 16:10:08 -0800 | [diff] [blame] | 193 | write_to_bazelrc 'build --define with_jemalloc=true' |
Jonathan Hseu | 83c6e0c | 2017-01-11 16:39:35 -0800 | [diff] [blame] | 194 | fi |
| 195 | |
Martin Wicke | bc456e3 | 2017-03-23 12:31:16 -0800 | [diff] [blame] | 196 | while [[ "$TF_NEED_GCP" == "" ]]; do |
A. Unique TensorFlower | 95a954a | 2016-12-28 13:40:08 -0800 | [diff] [blame] | 197 | read -p "Do you wish to build TensorFlow with "\ |
| 198 | "Google Cloud Platform support? [y/N] " INPUT |
| 199 | case $INPUT in |
| 200 | [Yy]* ) echo "Google Cloud Platform support will be enabled for "\ |
| 201 | "TensorFlow"; TF_NEED_GCP=1;; |
| 202 | [Nn]* ) echo "No Google Cloud Platform support will be enabled for "\ |
| 203 | "TensorFlow"; TF_NEED_GCP=0;; |
| 204 | "" ) echo "No Google Cloud Platform support will be enabled for "\ |
| 205 | "TensorFlow"; TF_NEED_GCP=0;; |
| 206 | * ) echo "Invalid selection: " $INPUT;; |
| 207 | esac |
| 208 | done |
| 209 | |
Martin Wicke | bc456e3 | 2017-03-23 12:31:16 -0800 | [diff] [blame] | 210 | if [[ "$TF_NEED_GCP" == "1" ]]; then |
A. Unique TensorFlower | ccbc899 | 2017-04-04 16:10:08 -0800 | [diff] [blame] | 211 | write_to_bazelrc 'build --define with_gcp_support=true' |
A. Unique TensorFlower | 95a954a | 2016-12-28 13:40:08 -0800 | [diff] [blame] | 212 | fi |
| 213 | |
Martin Wicke | bc456e3 | 2017-03-23 12:31:16 -0800 | [diff] [blame] | 214 | while [[ "$TF_NEED_HDFS" == "" ]]; do |
Jonathan Hseu | 9f5b098 | 2016-09-19 11:14:58 -0800 | [diff] [blame] | 215 | read -p "Do you wish to build TensorFlow with "\ |
| 216 | "Hadoop File System support? [y/N] " INPUT |
| 217 | case $INPUT in |
| 218 | [Yy]* ) echo "Hadoop File System support will be enabled for "\ |
| 219 | "TensorFlow"; TF_NEED_HDFS=1;; |
| 220 | [Nn]* ) echo "No Hadoop File System support will be enabled for "\ |
| 221 | "TensorFlow"; TF_NEED_HDFS=0;; |
| 222 | "" ) echo "No Hadoop File System support will be enabled for "\ |
| 223 | "TensorFlow"; TF_NEED_HDFS=0;; |
| 224 | * ) echo "Invalid selection: " $INPUT;; |
| 225 | esac |
| 226 | done |
| 227 | |
Martin Wicke | bc456e3 | 2017-03-23 12:31:16 -0800 | [diff] [blame] | 228 | if [[ "$TF_NEED_HDFS" == "1" ]]; then |
A. Unique TensorFlower | ccbc899 | 2017-04-04 16:10:08 -0800 | [diff] [blame] | 229 | write_to_bazelrc 'build --define with_hdfs_support=true' |
Jonathan Hseu | 9f5b098 | 2016-09-19 11:14:58 -0800 | [diff] [blame] | 230 | fi |
| 231 | |
Peter Hawkins | 1e67c90 | 2017-01-09 12:04:37 -0800 | [diff] [blame] | 232 | ## Enable XLA. |
Martin Wicke | bc456e3 | 2017-03-23 12:31:16 -0800 | [diff] [blame] | 233 | while [[ "$TF_ENABLE_XLA" == "" ]]; do |
Peter Hawkins | 1e67c90 | 2017-01-09 12:04:37 -0800 | [diff] [blame] | 234 | read -p "Do you wish to build TensorFlow with the XLA just-in-time compiler (experimental)? [y/N] " INPUT |
| 235 | case $INPUT in |
| 236 | [Yy]* ) echo "XLA JIT support will be enabled for TensorFlow"; TF_ENABLE_XLA=1;; |
| 237 | [Nn]* ) echo "No XLA JIT support will be enabled for TensorFlow"; TF_ENABLE_XLA=0;; |
| 238 | "" ) echo "No XLA support will be enabled for TensorFlow"; TF_ENABLE_XLA=0;; |
| 239 | * ) echo "Invalid selection: " $INPUT;; |
| 240 | esac |
| 241 | done |
| 242 | |
Martin Wicke | bc456e3 | 2017-03-23 12:31:16 -0800 | [diff] [blame] | 243 | if [[ "$TF_ENABLE_XLA" == "1" ]]; then |
A. Unique TensorFlower | ccbc899 | 2017-04-04 16:10:08 -0800 | [diff] [blame] | 244 | write_to_bazelrc 'build --define with_xla_support=true' |
Peter Hawkins | 1e67c90 | 2017-01-09 12:04:37 -0800 | [diff] [blame] | 245 | fi |
| 246 | |
| 247 | |
Vijay Vasudevan | bf6b536 | 2015-12-02 15:04:40 -0800 | [diff] [blame] | 248 | # Invoke python_config and set up symlinks to python includes |
A. Unique TensorFlower | 8e0d6f1 | 2017-04-21 06:35:18 -0800 | [diff] [blame] | 249 | ./util/python/python_config.sh "$PYTHON_BIN_PATH" |
Vijay Vasudevan | bf6b536 | 2015-12-02 15:04:40 -0800 | [diff] [blame] | 250 | |
Martin Wicke | c4e3d4a | 2017-01-13 12:20:42 -0800 | [diff] [blame] | 251 | # Append CC optimization flags to bazel.rc |
| 252 | echo >> tools/bazel.rc |
| 253 | for opt in $CC_OPT_FLAGS; do |
Gunhan Gulsoy | acdbd68 | 2017-01-16 01:21:51 -0800 | [diff] [blame] | 254 | echo "build:opt --cxxopt=$opt --copt=$opt" >> tools/bazel.rc |
Martin Wicke | c4e3d4a | 2017-01-13 12:20:42 -0800 | [diff] [blame] | 255 | done |
| 256 | |
Andrew Selle | 09045e4 | 2016-09-06 08:19:04 -0800 | [diff] [blame] | 257 | # Run the gen_git_source to create links where bazel can track dependencies for |
| 258 | # git hash propagation |
| 259 | GEN_GIT_SOURCE=tensorflow/tools/git/gen_git_source.py |
| 260 | chmod a+x ${GEN_GIT_SOURCE} |
Patrick Nguyen | c5ab3dd | 2016-10-20 12:09:18 -0800 | [diff] [blame] | 261 | "${PYTHON_BIN_PATH}" ${GEN_GIT_SOURCE} --configure "${SOURCE_BASE_DIR}" |
Andrew Selle | 09045e4 | 2016-09-06 08:19:04 -0800 | [diff] [blame] | 262 | |
Benoit Steiner | a771598 | 2016-11-09 13:14:03 -0800 | [diff] [blame] | 263 | ## Set up SYCL-related environment settings |
| 264 | while [ "$TF_NEED_OPENCL" == "" ]; do |
| 265 | read -p "Do you wish to build TensorFlow with OpenCL support? [y/N] " INPUT |
| 266 | case $INPUT in |
| 267 | [Yy]* ) echo "OpenCL support will be enabled for TensorFlow"; TF_NEED_OPENCL=1;; |
| 268 | [Nn]* ) echo "No OpenCL support will be enabled for TensorFlow"; TF_NEED_OPENCL=0;; |
| 269 | "" ) echo "No OpenCL support will be enabled for TensorFlow"; TF_NEED_OPENCL=0;; |
| 270 | * ) echo "Invalid selection: " $INPUT;; |
| 271 | esac |
| 272 | done |
| 273 | |
Manjunath Kudlur | f41959c | 2015-11-06 16:27:58 -0800 | [diff] [blame] | 274 | ## Set up Cuda-related environment settings |
| 275 | |
| 276 | while [ "$TF_NEED_CUDA" == "" ]; do |
Andrew Harp | 1cb9689 | 2016-12-08 20:05:49 -0800 | [diff] [blame] | 277 | read -p "Do you wish to build TensorFlow with CUDA support? [y/N] " INPUT |
Manjunath Kudlur | f41959c | 2015-11-06 16:27:58 -0800 | [diff] [blame] | 278 | case $INPUT in |
Andrew Harp | 1cb9689 | 2016-12-08 20:05:49 -0800 | [diff] [blame] | 279 | [Yy]* ) echo "CUDA support will be enabled for TensorFlow"; TF_NEED_CUDA=1;; |
| 280 | [Nn]* ) echo "No CUDA support will be enabled for TensorFlow"; TF_NEED_CUDA=0;; |
| 281 | "" ) echo "No CUDA support will be enabled for TensorFlow"; TF_NEED_CUDA=0;; |
Manjunath Kudlur | f41959c | 2015-11-06 16:27:58 -0800 | [diff] [blame] | 282 | * ) echo "Invalid selection: " $INPUT;; |
| 283 | esac |
| 284 | done |
| 285 | |
A. Unique TensorFlower | 2c598e8 | 2016-08-25 10:22:44 -0800 | [diff] [blame] | 286 | export TF_NEED_CUDA |
A. Unique TensorFlower | ccbc899 | 2017-04-04 16:10:08 -0800 | [diff] [blame] | 287 | write_action_env_to_bazelrc "TF_NEED_CUDA" "$TF_NEED_CUDA" |
A. Unique TensorFlower | 8d393ea | 2017-03-30 07:38:55 -0800 | [diff] [blame] | 288 | |
Rohan Jain | aab0997 | 2017-01-05 14:39:17 -0800 | [diff] [blame] | 289 | export TF_NEED_OPENCL |
A. Unique TensorFlower | 8268476 | 2017-04-05 11:30:37 -0800 | [diff] [blame] | 290 | write_action_env_to_bazelrc "TF_NEED_OPENCL" "$TF_NEED_OPENCL" |
A. Unique TensorFlower | 8d393ea | 2017-03-30 07:38:55 -0800 | [diff] [blame] | 291 | |
Benoit Steiner | a771598 | 2016-11-09 13:14:03 -0800 | [diff] [blame] | 292 | if [ "$TF_NEED_CUDA" == "1" ]; then |
A. Unique TensorFlower | 8d393ea | 2017-03-30 07:38:55 -0800 | [diff] [blame] | 293 | while [[ "$TF_CUDA_CLANG" == "" ]]; do |
| 294 | read -p "Do you want to use clang as CUDA compiler? [y/N] " INPUT |
| 295 | case $INPUT in |
| 296 | [Yy]* ) echo "Clang will be used as CUDA compiler"; TF_CUDA_CLANG=1;; |
| 297 | [Nn]* ) echo "nvcc will be used as CUDA compiler"; TF_CUDA_CLANG=0;; |
| 298 | "" ) echo "nvcc will be used as CUDA compiler"; TF_CUDA_CLANG=0;; |
| 299 | * ) echo "Invalid selection: " $INPUT;; |
| 300 | esac |
| 301 | done |
| 302 | |
| 303 | export TF_CUDA_CLANG |
A. Unique TensorFlower | ccbc899 | 2017-04-04 16:10:08 -0800 | [diff] [blame] | 304 | write_action_env_to_bazelrc "TF_CUDA_CLANG" "$TF_CUDA_CLANG" |
A. Unique TensorFlower | 8d393ea | 2017-03-30 07:38:55 -0800 | [diff] [blame] | 305 | |
Vijay Vasudevan | 80a5a3e | 2016-03-29 18:23:11 -0800 | [diff] [blame] | 306 | # Set up which gcc nvcc should use as the host compiler |
Andrew Harp | 1cb9689 | 2016-12-08 20:05:49 -0800 | [diff] [blame] | 307 | # No need to set this on Windows |
A. Unique TensorFlower | 8d393ea | 2017-03-30 07:38:55 -0800 | [diff] [blame] | 308 | while [[ "$TF_CUDA_CLANG" != "1" ]] && ! is_windows && true; do |
Vijay Vasudevan | 80a5a3e | 2016-03-29 18:23:11 -0800 | [diff] [blame] | 309 | fromuser="" |
| 310 | if [ -z "$GCC_HOST_COMPILER_PATH" ]; then |
A. Unique TensorFlower | edaf3b3 | 2016-10-10 10:26:22 -0800 | [diff] [blame] | 311 | default_gcc_host_compiler_path=$(which gcc || true) |
Shanqing Cai | a81c4f9 | 2016-07-22 10:37:35 -0800 | [diff] [blame] | 312 | read -p "Please specify which gcc should be used by nvcc as the host compiler. [Default is $default_gcc_host_compiler_path]: " GCC_HOST_COMPILER_PATH |
Vijay Vasudevan | 80a5a3e | 2016-03-29 18:23:11 -0800 | [diff] [blame] | 313 | fromuser="1" |
| 314 | if [ -z "$GCC_HOST_COMPILER_PATH" ]; then |
Andrew Harp | 1cb9689 | 2016-12-08 20:05:49 -0800 | [diff] [blame] | 315 | GCC_HOST_COMPILER_PATH="$default_gcc_host_compiler_path" |
Vijay Vasudevan | 80a5a3e | 2016-03-29 18:23:11 -0800 | [diff] [blame] | 316 | fi |
| 317 | fi |
| 318 | if [ -e "$GCC_HOST_COMPILER_PATH" ]; then |
A. Unique TensorFlower | b0bdff4 | 2016-08-26 12:50:48 -0800 | [diff] [blame] | 319 | export GCC_HOST_COMPILER_PATH |
A. Unique TensorFlower | ccbc899 | 2017-04-04 16:10:08 -0800 | [diff] [blame] | 320 | write_action_env_to_bazelrc "GCC_HOST_COMPILER_PATH" "$GCC_HOST_COMPILER_PATH" |
Vijay Vasudevan | 80a5a3e | 2016-03-29 18:23:11 -0800 | [diff] [blame] | 321 | break |
| 322 | fi |
| 323 | echo "Invalid gcc path. ${GCC_HOST_COMPILER_PATH} cannot be found" 1>&2 |
| 324 | if [ -z "$fromuser" ]; then |
| 325 | exit 1 |
| 326 | fi |
| 327 | GCC_HOST_COMPILER_PATH="" |
| 328 | # Retry |
| 329 | done |
| 330 | |
A. Unique TensorFlower | 8d393ea | 2017-03-30 07:38:55 -0800 | [diff] [blame] | 331 | # Set up which clang we should use as the cuda / host compiler. |
| 332 | while [[ "$TF_CUDA_CLANG" == "1" ]] && true; do |
| 333 | fromuser="" |
| 334 | if [ -z "$CLANG_CUDA_COMPILER_PATH" ]; then |
| 335 | default_clang_host_compiler_path=$(which clang || true) |
| 336 | read -p "Please specify which clang should be used as device and host compiler. [Default is $default_clang_host_compiler_path]: " CLANG_CUDA_COMPILER_PATH |
| 337 | fromuser="1" |
| 338 | if [ -z "$CLANG_CUDA_COMPILER_PATH" ]; then |
| 339 | CLANG_CUDA_COMPILER_PATH="$default_clang_host_compiler_path" |
| 340 | fi |
| 341 | fi |
| 342 | if [ -e "$CLANG_CUDA_COMPILER_PATH" ]; then |
| 343 | export CLANG_CUDA_COMPILER_PATH |
A. Unique TensorFlower | ccbc899 | 2017-04-04 16:10:08 -0800 | [diff] [blame] | 344 | write_action_env_to_bazelrc "CLANG_CUDA_COMPILER_PATH" "$CLANG_CUDA_COMPILER_PATH" |
A. Unique TensorFlower | 8d393ea | 2017-03-30 07:38:55 -0800 | [diff] [blame] | 345 | break |
| 346 | fi |
| 347 | echo "Invalid clang path. ${CLANG_CUDA_COMPILER_PATH} cannot be found" 1>&2 |
| 348 | if [ -z "$fromuser" ]; then |
| 349 | exit 1 |
| 350 | fi |
| 351 | CLANG_CUDA_COMPILER_PATH="" |
| 352 | # Retry |
| 353 | done |
| 354 | |
Manjunath Kudlur | f41959c | 2015-11-06 16:27:58 -0800 | [diff] [blame] | 355 | # Find out where the CUDA toolkit is installed |
| 356 | while true; do |
A. Unique TensorFlower | 8a59748 | 2016-01-29 09:34:18 -0800 | [diff] [blame] | 357 | # Configure the Cuda SDK version to use. |
A. Unique TensorFlower | 8a59748 | 2016-01-29 09:34:18 -0800 | [diff] [blame] | 358 | if [ -z "$TF_CUDA_VERSION" ]; then |
Andrew Harp | 1cb9689 | 2016-12-08 20:05:49 -0800 | [diff] [blame] | 359 | read -p "Please specify the CUDA SDK version you want to use, e.g. 7.0. [Leave empty to use system default]: " TF_CUDA_VERSION |
A. Unique TensorFlower | 8a59748 | 2016-01-29 09:34:18 -0800 | [diff] [blame] | 360 | fi |
| 361 | |
Manjunath Kudlur | f41959c | 2015-11-06 16:27:58 -0800 | [diff] [blame] | 362 | fromuser="" |
| 363 | if [ -z "$CUDA_TOOLKIT_PATH" ]; then |
| 364 | default_cuda_path=/usr/local/cuda |
Andrew Harp | 1cb9689 | 2016-12-08 20:05:49 -0800 | [diff] [blame] | 365 | if is_windows; then |
| 366 | if [ -z "$CUDA_PATH" ]; then |
| 367 | default_cuda_path="C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v8.0" |
| 368 | else |
| 369 | default_cuda_path="$(cygpath -m "$CUDA_PATH")" |
| 370 | fi |
| 371 | fi |
A. Unique TensorFlower | 8a59748 | 2016-01-29 09:34:18 -0800 | [diff] [blame] | 372 | read -p "Please specify the location where CUDA $TF_CUDA_VERSION toolkit is installed. Refer to README.md for more details. [Default is $default_cuda_path]: " CUDA_TOOLKIT_PATH |
Manjunath Kudlur | f41959c | 2015-11-06 16:27:58 -0800 | [diff] [blame] | 373 | fromuser="1" |
| 374 | if [ -z "$CUDA_TOOLKIT_PATH" ]; then |
Andrew Harp | 1cb9689 | 2016-12-08 20:05:49 -0800 | [diff] [blame] | 375 | CUDA_TOOLKIT_PATH="$default_cuda_path" |
Manjunath Kudlur | f41959c | 2015-11-06 16:27:58 -0800 | [diff] [blame] | 376 | fi |
| 377 | fi |
A. Unique TensorFlower | 8bf6ef1 | 2016-05-05 08:36:05 -0800 | [diff] [blame] | 378 | |
Vijay Vasudevan | fe056f0 | 2016-02-17 11:42:30 -0800 | [diff] [blame] | 379 | if [[ -z "$TF_CUDA_VERSION" ]]; then |
| 380 | TF_CUDA_EXT="" |
| 381 | else |
| 382 | TF_CUDA_EXT=".$TF_CUDA_VERSION" |
| 383 | fi |
A. Unique TensorFlower | 8bf6ef1 | 2016-05-05 08:36:05 -0800 | [diff] [blame] | 384 | |
Andrew Harp | 1cb9689 | 2016-12-08 20:05:49 -0800 | [diff] [blame] | 385 | if is_windows; then |
| 386 | CUDA_RT_LIB_PATH="lib/x64/cudart.lib" |
Jonathan Hseu | c058a01 | 2017-01-17 15:06:43 -0800 | [diff] [blame] | 387 | elif is_linux; then |
A. Unique TensorFlower | 8bf6ef1 | 2016-05-05 08:36:05 -0800 | [diff] [blame] | 388 | CUDA_RT_LIB_PATH="lib64/libcudart.so${TF_CUDA_EXT}" |
Jonathan Hseu | c058a01 | 2017-01-17 15:06:43 -0800 | [diff] [blame] | 389 | elif is_macos; then |
A. Unique TensorFlower | 8bf6ef1 | 2016-05-05 08:36:05 -0800 | [diff] [blame] | 390 | CUDA_RT_LIB_PATH="lib/libcudart${TF_CUDA_EXT}.dylib" |
| 391 | fi |
| 392 | |
| 393 | if [ -e "${CUDA_TOOLKIT_PATH}/${CUDA_RT_LIB_PATH}" ]; then |
A. Unique TensorFlower | 2c598e8 | 2016-08-25 10:22:44 -0800 | [diff] [blame] | 394 | export CUDA_TOOLKIT_PATH |
A. Unique TensorFlower | ccbc899 | 2017-04-04 16:10:08 -0800 | [diff] [blame] | 395 | write_action_env_to_bazelrc "CUDA_TOOLKIT_PATH" "$CUDA_TOOLKIT_PATH" |
A. Unique TensorFlower | b0bdff4 | 2016-08-26 12:50:48 -0800 | [diff] [blame] | 396 | export TF_CUDA_VERSION |
A. Unique TensorFlower | ccbc899 | 2017-04-04 16:10:08 -0800 | [diff] [blame] | 397 | write_action_env_to_bazelrc "TF_CUDA_VERSION" "$TF_CUDA_VERSION" |
Manjunath Kudlur | f41959c | 2015-11-06 16:27:58 -0800 | [diff] [blame] | 398 | break |
| 399 | fi |
A. Unique TensorFlower | 8bf6ef1 | 2016-05-05 08:36:05 -0800 | [diff] [blame] | 400 | echo "Invalid path to CUDA $TF_CUDA_VERSION toolkit. ${CUDA_TOOLKIT_PATH}/${CUDA_RT_LIB_PATH} cannot be found" |
| 401 | |
Manjunath Kudlur | f41959c | 2015-11-06 16:27:58 -0800 | [diff] [blame] | 402 | if [ -z "$fromuser" ]; then |
| 403 | exit 1 |
| 404 | fi |
Vijay Vasudevan | fe056f0 | 2016-02-17 11:42:30 -0800 | [diff] [blame] | 405 | # Retry |
A. Unique TensorFlower | 8a59748 | 2016-01-29 09:34:18 -0800 | [diff] [blame] | 406 | TF_CUDA_VERSION="" |
Manjunath Kudlur | f41959c | 2015-11-06 16:27:58 -0800 | [diff] [blame] | 407 | CUDA_TOOLKIT_PATH="" |
Manjunath Kudlur | f41959c | 2015-11-06 16:27:58 -0800 | [diff] [blame] | 408 | done |
| 409 | |
Martin Wicke | 916776a | 2016-01-14 07:30:00 -0800 | [diff] [blame] | 410 | # Find out where the cuDNN library is installed |
Manjunath Kudlur | f41959c | 2015-11-06 16:27:58 -0800 | [diff] [blame] | 411 | while true; do |
A. Unique TensorFlower | ccbc899 | 2017-04-04 16:10:08 -0800 | [diff] [blame] | 412 | # Configure the cuDNN version to use. |
A. Unique TensorFlower | 8a59748 | 2016-01-29 09:34:18 -0800 | [diff] [blame] | 413 | if [ -z "$TF_CUDNN_VERSION" ]; then |
A. Unique TensorFlower | ccbc899 | 2017-04-04 16:10:08 -0800 | [diff] [blame] | 414 | read -p "Please specify the cuDNN version you want to use. [Leave empty to use system default]: " TF_CUDNN_VERSION |
A. Unique TensorFlower | 8a59748 | 2016-01-29 09:34:18 -0800 | [diff] [blame] | 415 | fi |
| 416 | |
Manjunath Kudlur | f41959c | 2015-11-06 16:27:58 -0800 | [diff] [blame] | 417 | fromuser="" |
| 418 | if [ -z "$CUDNN_INSTALL_PATH" ]; then |
| 419 | default_cudnn_path=${CUDA_TOOLKIT_PATH} |
A. Unique TensorFlower | 8a59748 | 2016-01-29 09:34:18 -0800 | [diff] [blame] | 420 | read -p "Please specify the location where cuDNN $TF_CUDNN_VERSION library is installed. Refer to README.md for more details. [Default is $default_cudnn_path]: " CUDNN_INSTALL_PATH |
Manjunath Kudlur | f41959c | 2015-11-06 16:27:58 -0800 | [diff] [blame] | 421 | fromuser="1" |
| 422 | if [ -z "$CUDNN_INSTALL_PATH" ]; then |
| 423 | CUDNN_INSTALL_PATH=$default_cudnn_path |
| 424 | fi |
| 425 | # Result returned from "read" will be used unexpanded. That make "~" unuseable. |
| 426 | # Going through one more level of expansion to handle that. |
Andrew Harp | 1cb9689 | 2016-12-08 20:05:49 -0800 | [diff] [blame] | 427 | CUDNN_INSTALL_PATH=`"${PYTHON_BIN_PATH}" -c "import os; print(os.path.realpath(os.path.expanduser('${CUDNN_INSTALL_PATH}')))"` |
Manjunath Kudlur | f41959c | 2015-11-06 16:27:58 -0800 | [diff] [blame] | 428 | fi |
A. Unique TensorFlower | 8bf6ef1 | 2016-05-05 08:36:05 -0800 | [diff] [blame] | 429 | |
Vijay Vasudevan | fe056f0 | 2016-02-17 11:42:30 -0800 | [diff] [blame] | 430 | if [[ -z "$TF_CUDNN_VERSION" ]]; then |
| 431 | TF_CUDNN_EXT="" |
| 432 | else |
Benoit Steiner | 639b4e7 | 2017-02-08 09:25:09 -0800 | [diff] [blame] | 433 | TF_CUDNN_EXT=".$TF_CUDNN_VERSION" |
Vijay Vasudevan | fe056f0 | 2016-02-17 11:42:30 -0800 | [diff] [blame] | 434 | fi |
A. Unique TensorFlower | 8bf6ef1 | 2016-05-05 08:36:05 -0800 | [diff] [blame] | 435 | |
Andrew Harp | 1cb9689 | 2016-12-08 20:05:49 -0800 | [diff] [blame] | 436 | if is_windows; then |
| 437 | CUDA_DNN_LIB_PATH="lib/x64/cudnn.lib" |
| 438 | CUDA_DNN_LIB_ALT_PATH="lib/x64/cudnn.lib" |
Jonathan Hseu | c058a01 | 2017-01-17 15:06:43 -0800 | [diff] [blame] | 439 | elif is_linux; then |
A. Unique TensorFlower | 8bf6ef1 | 2016-05-05 08:36:05 -0800 | [diff] [blame] | 440 | CUDA_DNN_LIB_PATH="lib64/libcudnn.so${TF_CUDNN_EXT}" |
| 441 | CUDA_DNN_LIB_ALT_PATH="libcudnn.so${TF_CUDNN_EXT}" |
Jonathan Hseu | c058a01 | 2017-01-17 15:06:43 -0800 | [diff] [blame] | 442 | elif is_macos; then |
Benoit Steiner | 639b4e7 | 2017-02-08 09:25:09 -0800 | [diff] [blame] | 443 | CUDA_DNN_LIB_PATH="lib/libcudnn${TF_CUDNN_EXT}.dylib" |
| 444 | CUDA_DNN_LIB_ALT_PATH="libcudnn${TF_CUDNN_EXT}.dylib" |
A. Unique TensorFlower | 8bf6ef1 | 2016-05-05 08:36:05 -0800 | [diff] [blame] | 445 | fi |
| 446 | |
| 447 | if [ -e "$CUDNN_INSTALL_PATH/${CUDA_DNN_LIB_ALT_PATH}" -o -e "$CUDNN_INSTALL_PATH/${CUDA_DNN_LIB_PATH}" ]; then |
A. Unique TensorFlower | b0bdff4 | 2016-08-26 12:50:48 -0800 | [diff] [blame] | 448 | export TF_CUDNN_VERSION |
A. Unique TensorFlower | ccbc899 | 2017-04-04 16:10:08 -0800 | [diff] [blame] | 449 | write_action_env_to_bazelrc "TF_CUDNN_VERSION" "$TF_CUDNN_VERSION" |
A. Unique TensorFlower | 2c598e8 | 2016-08-25 10:22:44 -0800 | [diff] [blame] | 450 | export CUDNN_INSTALL_PATH |
A. Unique TensorFlower | ccbc899 | 2017-04-04 16:10:08 -0800 | [diff] [blame] | 451 | write_action_env_to_bazelrc "CUDNN_INSTALL_PATH" "$CUDNN_INSTALL_PATH" |
Eugene Brevdo | 56f1d64 | 2016-03-10 17:18:30 -0800 | [diff] [blame] | 452 | break |
| 453 | fi |
A. Unique TensorFlower | 8bf6ef1 | 2016-05-05 08:36:05 -0800 | [diff] [blame] | 454 | |
Jonathan Hseu | c058a01 | 2017-01-17 15:06:43 -0800 | [diff] [blame] | 455 | if is_linux; then |
Vijay Vasudevan | 93a975e | 2017-02-17 17:05:49 -0800 | [diff] [blame] | 456 | if ! type ldconfig > /dev/null 2>&1; then |
| 457 | LDCONFIG_BIN=/sbin/ldconfig |
| 458 | else |
| 459 | LDCONFIG_BIN=ldconfig |
| 460 | fi |
| 461 | CUDNN_PATH_FROM_LDCONFIG="$($LDCONFIG_BIN -p | sed -n 's/.*libcudnn.so .* => \(.*\)/\1/p')" |
A. Unique TensorFlower | 8bf6ef1 | 2016-05-05 08:36:05 -0800 | [diff] [blame] | 462 | if [ -e "${CUDNN_PATH_FROM_LDCONFIG}${TF_CUDNN_EXT}" ]; then |
A. Unique TensorFlower | b0bdff4 | 2016-08-26 12:50:48 -0800 | [diff] [blame] | 463 | export TF_CUDNN_VERSION |
A. Unique TensorFlower | ccbc899 | 2017-04-04 16:10:08 -0800 | [diff] [blame] | 464 | write_action_env_to_bazelrc "TF_CUDNN_VERSION" "$TF_CUDNN_VERSION" |
A. Unique TensorFlower | 2c598e8 | 2016-08-25 10:22:44 -0800 | [diff] [blame] | 465 | export CUDNN_INSTALL_PATH="$(dirname ${CUDNN_PATH_FROM_LDCONFIG})" |
A. Unique TensorFlower | ccbc899 | 2017-04-04 16:10:08 -0800 | [diff] [blame] | 466 | write_action_env_to_bazelrc "CUDNN_INSTALL_PATH" "$CUDNN_INSTALL_PATH" |
A. Unique TensorFlower | 8bf6ef1 | 2016-05-05 08:36:05 -0800 | [diff] [blame] | 467 | break |
| 468 | fi |
Manjunath Kudlur | f41959c | 2015-11-06 16:27:58 -0800 | [diff] [blame] | 469 | fi |
A. Unique TensorFlower | 8bf6ef1 | 2016-05-05 08:36:05 -0800 | [diff] [blame] | 470 | echo "Invalid path to cuDNN ${CUDNN_VERSION} toolkit. Neither of the following two files can be found:" |
| 471 | echo "${CUDNN_INSTALL_PATH}/${CUDA_DNN_LIB_PATH}" |
| 472 | echo "${CUDNN_INSTALL_PATH}/${CUDA_DNN_LIB_ALT_PATH}" |
Jonathan Hseu | c058a01 | 2017-01-17 15:06:43 -0800 | [diff] [blame] | 473 | if is_linux; then |
A. Unique TensorFlower | 8bf6ef1 | 2016-05-05 08:36:05 -0800 | [diff] [blame] | 474 | echo "${CUDNN_PATH_FROM_LDCONFIG}${TF_CUDNN_EXT}" |
| 475 | fi |
| 476 | |
Manjunath Kudlur | f41959c | 2015-11-06 16:27:58 -0800 | [diff] [blame] | 477 | if [ -z "$fromuser" ]; then |
| 478 | exit 1 |
| 479 | fi |
Vijay Vasudevan | fe056f0 | 2016-02-17 11:42:30 -0800 | [diff] [blame] | 480 | # Retry |
A. Unique TensorFlower | 8a59748 | 2016-01-29 09:34:18 -0800 | [diff] [blame] | 481 | TF_CUDNN_VERSION="" |
Manjunath Kudlur | f41959c | 2015-11-06 16:27:58 -0800 | [diff] [blame] | 482 | CUDNN_INSTALL_PATH="" |
Manjunath Kudlur | f41959c | 2015-11-06 16:27:58 -0800 | [diff] [blame] | 483 | done |
| 484 | |
Vijay Vasudevan | fe056f0 | 2016-02-17 11:42:30 -0800 | [diff] [blame] | 485 | # Configure the compute capabilities that TensorFlow builds for. |
| 486 | # Since Cuda toolkit is not backward-compatible, this is not guaranteed to work. |
| 487 | while true; do |
| 488 | fromuser="" |
A. Unique TensorFlower | 2c598e8 | 2016-08-25 10:22:44 -0800 | [diff] [blame] | 489 | default_cuda_compute_capabilities="3.5,5.2" |
Vijay Vasudevan | fe056f0 | 2016-02-17 11:42:30 -0800 | [diff] [blame] | 490 | if [ -z "$TF_CUDA_COMPUTE_CAPABILITIES" ]; then |
Vijay Vasudevan | 4dffee7 | 2015-11-12 11:27:00 -0800 | [diff] [blame] | 491 | cat << EOF |
| 492 | Please specify a list of comma-separated Cuda compute capabilities you want to build with. |
| 493 | You can find the compute capability of your device at: https://developer.nvidia.com/cuda-gpus. |
| 494 | Please note that each additional compute capability significantly increases your build time and binary size. |
| 495 | EOF |
Vijay Vasudevan | fe056f0 | 2016-02-17 11:42:30 -0800 | [diff] [blame] | 496 | read -p "[Default is: \"3.5,5.2\"]: " TF_CUDA_COMPUTE_CAPABILITIES |
| 497 | fromuser=1 |
| 498 | fi |
A. Unique TensorFlower | 2c598e8 | 2016-08-25 10:22:44 -0800 | [diff] [blame] | 499 | if [ -z "$TF_CUDA_COMPUTE_CAPABILITIES" ]; then |
| 500 | TF_CUDA_COMPUTE_CAPABILITIES=$default_cuda_compute_capabilities |
| 501 | fi |
Vijay Vasudevan | fe056f0 | 2016-02-17 11:42:30 -0800 | [diff] [blame] | 502 | # Check whether all capabilities from the input is valid |
| 503 | COMPUTE_CAPABILITIES=${TF_CUDA_COMPUTE_CAPABILITIES//,/ } |
| 504 | ALL_VALID=1 |
| 505 | for CAPABILITY in $COMPUTE_CAPABILITIES; do |
| 506 | if [[ ! "$CAPABILITY" =~ [0-9]+.[0-9]+ ]]; then |
| 507 | echo "Invalid compute capability: " $CAPABILITY |
| 508 | ALL_VALID=0 |
Vijay Vasudevan | 4dffee7 | 2015-11-12 11:27:00 -0800 | [diff] [blame] | 509 | break |
| 510 | fi |
Vijay Vasudevan | 4dffee7 | 2015-11-12 11:27:00 -0800 | [diff] [blame] | 511 | done |
Vijay Vasudevan | fe056f0 | 2016-02-17 11:42:30 -0800 | [diff] [blame] | 512 | if [ "$ALL_VALID" == "0" ]; then |
| 513 | if [ -z "$fromuser" ]; then |
| 514 | exit 1 |
| 515 | fi |
| 516 | else |
A. Unique TensorFlower | b0bdff4 | 2016-08-26 12:50:48 -0800 | [diff] [blame] | 517 | export TF_CUDA_COMPUTE_CAPABILITIES |
A. Unique TensorFlower | ccbc899 | 2017-04-04 16:10:08 -0800 | [diff] [blame] | 518 | write_action_env_to_bazelrc "TF_CUDA_COMPUTE_CAPABILITIES" "$TF_CUDA_COMPUTE_CAPABILITIES" |
Vijay Vasudevan | fe056f0 | 2016-02-17 11:42:30 -0800 | [diff] [blame] | 519 | break |
Vijay Vasudevan | 4dffee7 | 2015-11-12 11:27:00 -0800 | [diff] [blame] | 520 | fi |
Vijay Vasudevan | fe056f0 | 2016-02-17 11:42:30 -0800 | [diff] [blame] | 521 | TF_CUDA_COMPUTE_CAPABILITIES="" |
| 522 | done |
Vijay Vasudevan | 4dffee7 | 2015-11-12 11:27:00 -0800 | [diff] [blame] | 523 | |
Andrew Harp | 1cb9689 | 2016-12-08 20:05:49 -0800 | [diff] [blame] | 524 | if is_windows; then |
| 525 | # The following three variables are needed for MSVC toolchain configuration in Bazel |
| 526 | export CUDA_PATH="$CUDA_TOOLKIT_PATH" |
| 527 | export CUDA_COMPUTE_CAPABILITIES="$TF_CUDA_COMPUTE_CAPABILITIES" |
| 528 | export NO_WHOLE_ARCHIVE_OPTION=1 |
A. Unique TensorFlower | ccbc899 | 2017-04-04 16:10:08 -0800 | [diff] [blame] | 529 | write_action_env_to_bazelrc "CUDA_PATH" "$CUDA_PATH" |
| 530 | write_action_env_to_bazelrc "CUDA_COMPUTE_CAPABILITIES" "$CUDA_COMPUTE_CAPABILITIES" |
| 531 | write_action_env_to_bazelrc "NO_WHOLE_ARCHIVE_OPTION" "1" |
Andrew Harp | 1cb9689 | 2016-12-08 20:05:49 -0800 | [diff] [blame] | 532 | fi |
| 533 | |
Benoit Steiner | a771598 | 2016-11-09 13:14:03 -0800 | [diff] [blame] | 534 | # end of if "$TF_NEED_CUDA" == "1" |
| 535 | fi |
| 536 | |
| 537 | # OpenCL configuration |
| 538 | |
| 539 | if [ "$TF_NEED_OPENCL" == "1" ]; then |
| 540 | |
| 541 | # Determine which C++ compiler should be used as the host compiler |
| 542 | while true; do |
| 543 | fromuser="" |
| 544 | if [ -z "$HOST_CXX_COMPILER" ]; then |
Jonathan Hseu | bed8383 | 2016-12-22 15:38:30 -0800 | [diff] [blame] | 545 | default_cxx_host_compiler=$(which clang++-3.6 || true) |
Benoit Steiner | a771598 | 2016-11-09 13:14:03 -0800 | [diff] [blame] | 546 | read -p "Please specify which C++ compiler should be used as the host C++ compiler. [Default is $default_cxx_host_compiler]: " HOST_CXX_COMPILER |
| 547 | fromuser="1" |
| 548 | if [ -z "$HOST_CXX_COMPILER" ]; then |
| 549 | HOST_CXX_COMPILER=$default_cxx_host_compiler |
| 550 | fi |
| 551 | fi |
| 552 | if [ -e "$HOST_CXX_COMPILER" ]; then |
| 553 | export HOST_CXX_COMPILER |
A. Unique TensorFlower | 8268476 | 2017-04-05 11:30:37 -0800 | [diff] [blame] | 554 | write_action_env_to_bazelrc "HOST_CXX_COMPILER" "$HOST_CXX_COMPILER" |
Benoit Steiner | a771598 | 2016-11-09 13:14:03 -0800 | [diff] [blame] | 555 | break |
| 556 | fi |
| 557 | echo "Invalid C++ compiler path. ${HOST_CXX_COMPILER} cannot be found" 1>&2 |
| 558 | if [ -z "$fromuser" ]; then |
| 559 | exit 1 |
| 560 | fi |
| 561 | HOST_CXX_COMPILER="" |
| 562 | # Retry |
| 563 | done |
| 564 | |
| 565 | # Determine which C compiler should be used as the host compiler |
| 566 | while true; do |
| 567 | fromuser="" |
| 568 | if [ -z "$HOST_C_COMPILER" ]; then |
Jonathan Hseu | bed8383 | 2016-12-22 15:38:30 -0800 | [diff] [blame] | 569 | default_c_host_compiler=$(which clang-3.6 || true) |
Benoit Steiner | a771598 | 2016-11-09 13:14:03 -0800 | [diff] [blame] | 570 | read -p "Please specify which C compiler should be used as the host C compiler. [Default is $default_c_host_compiler]: " HOST_C_COMPILER |
| 571 | fromuser="1" |
| 572 | if [ -z "$HOST_C_COMPILER" ]; then |
| 573 | HOST_C_COMPILER=$default_c_host_compiler |
| 574 | fi |
| 575 | fi |
| 576 | if [ -e "$HOST_C_COMPILER" ]; then |
| 577 | export HOST_C_COMPILER |
A. Unique TensorFlower | 8268476 | 2017-04-05 11:30:37 -0800 | [diff] [blame] | 578 | write_action_env_to_bazelrc "HOST_C_COMPILER" "$HOST_C_COMPILER" |
Benoit Steiner | a771598 | 2016-11-09 13:14:03 -0800 | [diff] [blame] | 579 | break |
| 580 | fi |
| 581 | echo "Invalid C compiler path. ${HOST_C_COMPILER} cannot be found" 1>&2 |
| 582 | if [ -z "$fromuser" ]; then |
| 583 | exit 1 |
| 584 | fi |
| 585 | HOST_C_COMPILER="" |
| 586 | # Retry |
| 587 | done |
| 588 | |
| 589 | while true; do |
| 590 | # Configure the OPENCL version to use. |
| 591 | TF_OPENCL_VERSION="1.2" |
| 592 | |
| 593 | # Point to ComputeCpp root |
| 594 | if [ -z "$COMPUTECPP_TOOLKIT_PATH" ]; then |
| 595 | default_computecpp_toolkit_path=/usr/local/computecpp |
Martin Wicke | 2e4869a | 2016-12-14 15:46:53 -0800 | [diff] [blame] | 596 | read -p "Please specify the location where ComputeCpp for SYCL $TF_OPENCL_VERSION is installed. [Default is $default_computecpp_toolkit_path]: " COMPUTECPP_TOOLKIT_PATH |
Benoit Steiner | a771598 | 2016-11-09 13:14:03 -0800 | [diff] [blame] | 597 | fromuser="1" |
| 598 | if [ -z "$COMPUTECPP_TOOLKIT_PATH" ]; then |
| 599 | COMPUTECPP_TOOLKIT_PATH=$default_computecpp_toolkit_path |
| 600 | fi |
| 601 | fi |
| 602 | |
Jonathan Hseu | c058a01 | 2017-01-17 15:06:43 -0800 | [diff] [blame] | 603 | if is_linux; then |
Benoit Steiner | a771598 | 2016-11-09 13:14:03 -0800 | [diff] [blame] | 604 | SYCL_RT_LIB_PATH="lib/libComputeCpp.so" |
| 605 | fi |
| 606 | |
| 607 | if [ -e "${COMPUTECPP_TOOLKIT_PATH}/${SYCL_RT_LIB_PATH}" ]; then |
| 608 | export COMPUTECPP_TOOLKIT_PATH |
A. Unique TensorFlower | 8268476 | 2017-04-05 11:30:37 -0800 | [diff] [blame] | 609 | write_action_env_to_bazelrc "COMPUTECPP_TOOLKIT_PATH" "$COMPUTECPP_TOOLKIT_PATH" |
Benoit Steiner | a771598 | 2016-11-09 13:14:03 -0800 | [diff] [blame] | 610 | break |
| 611 | fi |
| 612 | echo "Invalid SYCL $TF_OPENCL_VERSION library path. ${COMPUTECPP_TOOLKIT_PATH}/${SYCL_RT_LIB_PATH} cannot be found" |
| 613 | |
| 614 | if [ -z "$fromuser" ]; then |
| 615 | exit 1 |
| 616 | fi |
| 617 | # Retry |
| 618 | TF_OPENCL_VERSION="" |
| 619 | COMPUTECPP_TOOLKIT_PATH="" |
| 620 | done |
| 621 | |
Benoit Steiner | a771598 | 2016-11-09 13:14:03 -0800 | [diff] [blame] | 622 | # end of if "$TF_NEED_OPENCL" == "1" |
| 623 | fi |
| 624 | |
A. Unique TensorFlower | ccbc899 | 2017-04-04 16:10:08 -0800 | [diff] [blame] | 625 | # TODO(gunan): Remove once bazel correctly handles changes in remote repositories. |
| 626 | bazel clean |
Manjunath Kudlur | f41959c | 2015-11-06 16:27:58 -0800 | [diff] [blame] | 627 | echo "Configuration finished" |