experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

load libcupti.so shared object if libcupti.so.10.1 is not installed. This makes it possible for tensorflow to load CUPTI 10.2 via libcupti.so symlink in the event that libcupti.so.10.1 is missing. 

PiperOrigin-RevId: 316782424
Change-Id: I930912f1d3a8fa80e8b91c41214c374650f08847
This commit is contained in:
A. Unique TensorFlower 2020-06-16 16:39:49 -07:00 committed by TensorFlower Gardener
parent 897e3c0eca
commit 267f956246
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/stream_executor/platform/default/dso_loader.cc
View File

@ -101,13 +101,11 @@ port::StatusOr<void*> GetCurandDsoHandle() {
}
port::StatusOr<void*> GetCuptiDsoHandle() {
#if defined(ANDROID_TEGRA)
// On Android devices the CUDA version number is not added to the library
// name.
// Load specific version of CUPTI this is built.
auto status_or_handle = GetDsoHandle("cupti", GetCudaVersion());
if (status_or_handle.ok()) return status_or_handle;
// Load whatever libcupti.so user specified.
return GetDsoHandle("cupti", "");
#else
return GetDsoHandle("cupti", GetCudaVersion());
#endif
}
port::StatusOr<void*> GetCudnnDsoHandle() {