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.

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