aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/stream_executor/dso_loader.cc
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2016-06-24 16:50:40 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-06-24 18:05:02 -0700
commit67324b1e3af826c4c491802f4022a5f5be9f6670 (patch)
tree018e21feade905a14d8beb2c8e7ebdfd905bbf51 /tensorflow/stream_executor/dso_loader.cc
parente30936c026655f1b2f4f45997da32c257d18b076 (diff)
Merge changes from github.
Change: 125835079
Diffstat (limited to 'tensorflow/stream_executor/dso_loader.cc')
-rw-r--r--tensorflow/stream_executor/dso_loader.cc2
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/stream_executor/dso_loader.cc b/tensorflow/stream_executor/dso_loader.cc
index fab3385cca..cce31ef4dc 100644
--- a/tensorflow/stream_executor/dso_loader.cc
+++ b/tensorflow/stream_executor/dso_loader.cc
@@ -72,7 +72,7 @@ string GetCudnnVersion() { return ""; }
}
/* static */ port::Status DsoLoader::GetLibcudaDsoHandle(void** dso_handle) {
- return GetDsoHandle(FindDsoPath(tensorflow::internal::FormatLibraryFileName("cuda", ""),
+ return GetDsoHandle(FindDsoPath(tensorflow::internal::FormatLibraryFileName("cuda", "1"),
GetCudaDriverLibraryPath()),
dso_handle);
}