aboutsummaryrefslogtreecommitdiffhomepage
path: root/third_party/gpus/cuda
diff options
context:
space:
mode:
authorGravatar Manjunath Kudlur <keveman@gmail.com>2015-11-20 10:30:41 -0800
committerGravatar Manjunath Kudlur <keveman@gmail.com>2015-11-20 10:30:41 -0800
commit9c3043ff3bf31a6a81810b4ce9e87ef936f1f529 (patch)
tree6cfbf1de2b9b58cceee59feecbb1ff5069b1e31a /third_party/gpus/cuda
parentf7918e1dcd5b0c1f8114f488fc35a63a81e94535 (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-xthird_party/gpus/cuda/cuda_config.sh25
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