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

Make TryDlopenCUDALibraries use the cached SO loaders

Browse Source 
This reduces the amount of logspam in OSS tests.

PiperOrigin-RevId: 350228487
Change-Id: I790cf62a1cdc7ae7be29984e1ef0a39d81830239
This commit is contained in:
Sanjoy Das 2021-01-05 15:08:29 -08:00 committed by TensorFlower Gardener
parent 0c79db02d5
commit f486e58bcf
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tensorflow/stream_executor/platform/default/dlopen_checker.cc
View File

@ -21,14 +21,16 @@ namespace internal {
namespace DsoLoader {
port::Status TryDlopenCUDALibraries() {
auto cudart_status = GetCudaRuntimeDsoHandle();
auto cublas_status = GetCublasDsoHandle();
auto cublaslt_status = GetCublasLtDsoHandle();
auto cufft_status = GetCufftDsoHandle();
auto curand_status = GetCurandDsoHandle();
auto cusolver_status = GetCusolverDsoHandle();
auto cusparse_status = GetCusparseDsoHandle();
auto cudnn_status = GetCudnnDsoHandle();
namespace CachedLoader = ::stream_executor::internal::CachedDsoLoader;
auto cudart_status = CachedLoader::GetCudaRuntimeDsoHandle();
auto cublas_status = CachedLoader::GetCublasDsoHandle();
auto cublaslt_status = CachedLoader::GetCublasLtDsoHandle();
auto cufft_status = CachedLoader::GetCufftDsoHandle();
auto curand_status = CachedLoader::GetCurandDsoHandle();
auto cusolver_status = CachedLoader::GetCusolverDsoHandle();
auto cusparse_status = CachedLoader::GetCusparseDsoHandle();
auto cudnn_status = CachedLoader::GetCudnnDsoHandle();
if (!cudart_status.status().ok() || !cublas_status.status().ok() ||
!cufft_status.status().ok() || !curand_status.status().ok() ||
!cusolver_status.status().ok() || !cusparse_status.status().ok() ||