aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/stream_executor/dso_loader.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/stream_executor/dso_loader.h')
-rw-r--r--tensorflow/stream_executor/dso_loader.h5
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);
};