diff options
| author |  A. Unique TensorFlower <nobody@tensorflow.org> | 2016-05-05 08:36:05 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-05-05 09:41:47 -0700 |
| commit | 8bf6ef1337359993a8be057c0dc90da8f5a6e4fa (patch) | |
| tree | c7367050bf36d6f4b17a93d06700dc7169012ac1 /tensorflow/stream_executor/dso_loader.h | |
| parent | 931e848c28e97e8cae410af242f8e09d75663ee4 (diff) | |
Merge changes from github.
Change: 121586635
Diffstat (limited to 'tensorflow/stream_executor/dso_loader.h')
| -rw-r--r-- | tensorflow/stream_executor/dso_loader.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/dso_loader.h b/tensorflow/stream_executor/dso_loader.h index ba1690c320..2afbc294df 100644 --- a/tensorflow/stream_executor/dso_loader.h +++ b/tensorflow/stream_executor/dso_loader.h @@ -91,6 +91,11 @@ class DsoLoader { static string FindDsoPath(port::StringPiece library_name, port::StringPiece runfiles_relpath); + // Return platform dependent paths for DSOs + static string GetCudaLibraryDirPath(); + static string GetCudaDriverLibraryPath(); + static string GetCudaCuptiLibraryPath(); + SE_DISALLOW_COPY_AND_ASSIGN(DsoLoader); }; |