aboutsummaryrefslogtreecommitdiffhomepage
path: root/tools
diff options
context:
space:
mode:
authorGravatar Manjunath Kudlur <keveman@gmail.com>2016-03-07 15:09:08 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-03-07 15:14:02 -0800
commit13024c58fa87a87db23f6869a0c33598bb5c027d (patch)
tree335e5de9418f0008fb2d95bb1cdf49b27688a630 /tools
parentd82cc200e027552d46f1a0ff193f9cd48992c3f0 (diff)
Changed the cuda_crosstool_condition to check for a define of using_cuda.
Checking for a specific crosstool_top directory doesn't work when TensorFlow is a sub-module for a different project. Change: 116592676
Diffstat (limited to 'tools')
-rw-r--r--tools/bazel.rc.template1
1 files changed, 1 insertions, 0 deletions
diff --git a/tools/bazel.rc.template b/tools/bazel.rc.template
index 5bdc9884e8..e90ec790fd 100644
--- a/tools/bazel.rc.template
+++ b/tools/bazel.rc.template
@@ -1,4 +1,5 @@
build:cuda --crosstool_top=//third_party/gpus/crosstool
+build:cuda --define=using_cuda=true
build --force_python=py$PYTHON_MAJOR_VERSION
build --python$PYTHON_MAJOR_VERSION_path=$PYTHON_BINARY