diff options
author | Kiril Gorovoy <kgorovoy@google.com> | 2016-04-07 15:00:19 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-04-07 16:06:47 -0700 |
commit | dbc7edb41702876251ac4f40f36f846e86692351 (patch) | |
tree | 4e01bc1b251f0deaf154404a3a091885f6e0623a /third_party/gpus | |
parent | 613fa16a620b56312bdee9888177d34daa1918e7 (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/BUILD | 4 |
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([ |