diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-05-17 07:45:25 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-05-17 07:49:02 -0700 |
commit | eb937e62c54ffaa607283cfa9db1359e9a0b62fa (patch) | |
tree | d790133851b4ad77d0e9b60e66d2c673afed3abe /third_party/gpus | |
parent | 16c202635e3cc18031078b82bca2089fc921213d (diff) |
Fixing issue with cuda configuration when cudnn.h located in /usr/include/cudnn.h
PiperOrigin-RevId: 156306372
Diffstat (limited to 'third_party/gpus')
-rw-r--r-- | third_party/gpus/cuda_configure.bzl | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/third_party/gpus/cuda_configure.bzl b/third_party/gpus/cuda_configure.bzl index 6994db0a04..c6deae05b8 100644 --- a/third_party/gpus/cuda_configure.bzl +++ b/third_party/gpus/cuda_configure.bzl @@ -787,7 +787,7 @@ def _symlink_genrule_for_dir(repository_ctx, src_dir, dest_dir, genrule_name, if dest_files[i] != "": # If we have only one file to link we do not want to use the dest_dir, as # $(@D) will include the full path to the file. - dest = ' $(@D)/' + dest_dir + dest_files[i] if len(dest_files) != 1 else ' $(@D)' + dest_files[i] + dest = ' $(@D)/' + dest_dir + dest_files[i] if len(dest_files) != 1 else ' $(@D)/' + dest_files[i] command.append('ln -s ' + src_files[i] + dest) outs.append(' "' + dest_dir + dest_files[i] + '",') genrule = _genrule(src_dir, genrule_name, " && ".join(command), @@ -875,8 +875,8 @@ def _create_cuda_repository(repository_ctx): included_files = _read_dir(repository_ctx, cuda_include_path).replace( cuda_include_path, '').splitlines() if '/cudnn.h' not in included_files: - genrules.append(_symlink_genrule_for_dir(repository_ctx, None, "", - "cudnn-include", [cudnn_header_dir + "/cudnn.h"], ["include/cudnn.h"])) + genrules.append(_symlink_genrule_for_dir(repository_ctx, None, "include/", + "cudnn-include", [cudnn_header_dir + "/cudnn.h"], ["cudnn.h"])) else: genrules.append( 'filegroup(\n' + |