diff options
author | Manjunath Kudlur <keveman@gmail.com> | 2015-11-20 10:30:41 -0800 |
---|---|---|
committer | Manjunath Kudlur <keveman@gmail.com> | 2015-11-20 10:30:41 -0800 |
commit | 9c3043ff3bf31a6a81810b4ce9e87ef936f1f529 (patch) | |
tree | 6cfbf1de2b9b58cceee59feecbb1ff5069b1e31a /third_party/gpus/cuda | |
parent | f7918e1dcd5b0c1f8114f488fc35a63a81e94535 (diff) |
TensorFlow: Improve performance of Alexnet
Changes:
* error message that refers to removed `DefaultSession` method.
* -Wnull-conversion warnings
* the "_start_time" attr for recvs when the flag "--brain_enable_scheduling_for_recvs" is set.
* typo in tutorial data download progress message.
* a typo ("however their installing"=>"however installing").
* typo, rename "TensorFlow Mechanics" to "How To" to be consistent with the website.
* a typo ("subtact"=>"subtract").
* protobuf examples in comments in tensorflow::Example.proto.
* formula formatting in MNIST beginner tutorial
* negative fraction-of-queue-full stats
* protobuf inclusion path so that Android demo will build under Blaze.
* small typo (moderatly > moderately)
* Session.run() to check that tensor arguments come from the session's graph.
* another six import
* seq2seq typo in bazel command
Base CL: 108349164
Diffstat (limited to 'third_party/gpus/cuda')
-rwxr-xr-x | third_party/gpus/cuda/cuda_config.sh | 25 |
1 files changed, 20 insertions, 5 deletions
diff --git a/third_party/gpus/cuda/cuda_config.sh b/third_party/gpus/cuda/cuda_config.sh index 034298590b..cde23664de 100755 --- a/third_party/gpus/cuda/cuda_config.sh +++ b/third_party/gpus/cuda/cuda_config.sh @@ -1,4 +1,19 @@ #!/bin/bash +# Copyright 2015 Google Inc. All Rights Reserved. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# ============================================================================== + # A simple script to configure the Cuda tree needed for the TensorFlow GPU # build. We need both Cuda toolkit 7.0 and Cudnn 6.5. @@ -20,7 +35,7 @@ EOF CHECK_ONLY=0 # Parse the arguments. Add more arguments as the "case" line when needed. -while [[ $# > 0 ]]; do +while [[ $# -gt 0 ]]; do argument="$1" shift case $argument in @@ -84,8 +99,8 @@ function CheckAndLinkToSrcTree { # Link the output file to the source tree, avoiding self links if they are # the same. This could happen if invoked from the source tree by accident. - if [ ! `readlink -f $PWD` == `readlink -f $OUTPUTDIR/third_party/gpus/cuda` ]; then - mkdir -p `dirname $OUTPUTDIR/third_party/gpus/cuda/$FILE` + if [ ! $(readlink -f $PWD) == $(readlink -f $OUTPUTDIR/third_party/gpus/cuda) ]; then + mkdir -p $(dirname $OUTPUTDIR/third_party/gpus/cuda/$FILE) ln -sf $PWD/$FILE $OUTPUTDIR/third_party/gpus/cuda/$FILE fi } @@ -135,9 +150,9 @@ function LinkOneDir { SRC_PREFIX=$1 DST_PREFIX=$2 SRC_DIR=$3 - DST_DIR=`echo $SRC_DIR | sed "s,^$SRC_PREFIX,$DST_PREFIX,"` + DST_DIR=$(echo $SRC_DIR | sed "s,^$SRC_PREFIX,$DST_PREFIX,") mkdir -p $DST_DIR - FILE_LIST=`find -L $SRC_DIR -maxdepth 1 -type f` + FILE_LIST=$(find -L $SRC_DIR -maxdepth 1 -type f) if test "$FILE_LIST" != ""; then ln -sf $FILE_LIST $DST_DIR/ || exit -1 fi |