Skip to content

Commit

Permalink
Use canonical cuda include directory path in CROSSTOOL. (tensorflow#4096
Browse files Browse the repository at this point in the history
  • Loading branch information
davidzchen authored and Vijay Vasudevan committed Oct 31, 2016
1 parent 396513e commit 3ccdb2b
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions third_party/gpus/cuda_configure.bzl
Original file line number Diff line number Diff line change
Expand Up @@ -141,11 +141,11 @@ def _cuda_toolkit_path(repository_ctx, cuda_version):
# <basedir>/cuda-<version> and the provided path is <basedir>/cuda.
version_suffixed = "%s-%s" % (cuda_toolkit_path, cuda_version)
if repository_ctx.path(version_suffixed).exists:
return version_suffixed
cuda_toolkit_path = version_suffixed
# Returns the non-versioned path if cuda version is not provided or if the
# installation does not use a cuda- directory, such as on ArchLinux where
# CUDA installs directly to /opt/cuda.
return cuda_toolkit_path
return str(repository_ctx.path(cuda_toolkit_path).realpath)


def _cudnn_install_basedir(repository_ctx):
Expand Down

0 comments on commit 3ccdb2b

Please sign in to comment.