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

Use TF_CUDA_LIB_VERSION for cusolver and cusparse. Looks like with this change

Browse Source 
dso_loader.cc can find these libraries with Cuda 10.1.

PiperOrigin-RevId: 246373082
This commit is contained in:
Anna R 2019-05-02 12:56:13 -07:00 committed by TensorFlower Gardener
parent c030f6da89
commit 73a5fa2ae1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -83,11 +83,11 @@ port::StatusOr<void*> GetCufftDsoHandle() {
} }
port::StatusOr<void*> GetCusolverDsoHandle() { port::StatusOr<void*> GetCusolverDsoHandle() {
return GetDsoHandle("cusolver", GetCudaVersion()); return GetDsoHandle("cusolver", GetCudaLibVersion());
} }
port::StatusOr<void*> GetCusparseDsoHandle() { port::StatusOr<void*> GetCusparseDsoHandle() {
return GetDsoHandle("cusparse", GetCudaVersion()); return GetDsoHandle("cusparse", GetCudaLibVersion());
} }
port::StatusOr<void*> GetCurandDsoHandle() { port::StatusOr<void*> GetCurandDsoHandle() {