aboutsummaryrefslogtreecommitdiffhomepage
path: root/third_party/gpus
diff options
context:
space:
mode:
authorGravatar Kiril Gorovoy <kgorovoy@google.com>2016-04-07 15:00:19 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-04-07 16:06:47 -0700
commitdbc7edb41702876251ac4f40f36f846e86692351 (patch)
tree4e01bc1b251f0deaf154404a3a091885f6e0623a /third_party/gpus
parent613fa16a620b56312bdee9888177d34daa1918e7 (diff)
Fix genrules that didn't work when TensorFlow was imported as a submodule and compiled with --config=cuda.
Change: 119318629
Diffstat (limited to 'third_party/gpus')
-rw-r--r--third_party/gpus/cuda/BUILD4
1 files changed, 2 insertions, 2 deletions
diff --git a/third_party/gpus/cuda/BUILD b/third_party/gpus/cuda/BUILD
index bf7ad7662a..697990330e 100644
--- a/third_party/gpus/cuda/BUILD
+++ b/third_party/gpus/cuda/BUILD
@@ -141,7 +141,7 @@ genrule(
],
cmd = if_cuda(
# Under cuda config, create all the symbolic links to the actual cuda files
- "OUTPUTDIR=`readlink -f $(@D)/../../..`; cd third_party/gpus/cuda; OUTPUTDIR=$$OUTPUTDIR ./cuda_config.sh --check;",
+ "OUTPUTDIR=`readlink -f $(@D)/../../..`; cd `dirname $(location :cuda_config.sh)`; OUTPUTDIR=$$OUTPUTDIR ./cuda_config.sh --check;",
# Under non-cuda config, create all dummy files to make the build go through
";".join([
@@ -167,7 +167,7 @@ genrule(
],
cmd = if_cuda(
# Under cuda config, create the symbolic link to the actual cuda.config
- "ln -sf `readlink -f third_party/gpus/cuda/cuda.config` $(@D)/;",
+ "configfile=$(location :cuda.config); ln -sf `readlink -f $${configfile#*/*/*/}` $(@D)/;",
# Under non-cuda config, create the dummy file
";".join([