From 73a5fa2ae1663420655eab03267c26554249aaf1 Mon Sep 17 00:00:00 2001
From: Anna R <annarev@google.com>
Date: Thu, 2 May 2019 12:56:13 -0700
Subject: [PATCH] Use TF_CUDA_LIB_VERSION for cusolver and cusparse. Looks like
 with this change dso_loader.cc can find these libraries with Cuda 10.1.

PiperOrigin-RevId: 246373082
---
 tensorflow/stream_executor/platform/default/dso_loader.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tensorflow/stream_executor/platform/default/dso_loader.cc b/tensorflow/stream_executor/platform/default/dso_loader.cc
index 54a032469d5..80d71e25d48 100644
--- a/tensorflow/stream_executor/platform/default/dso_loader.cc
+++ b/tensorflow/stream_executor/platform/default/dso_loader.cc
@@ -83,11 +83,11 @@ port::StatusOr<void*> GetCufftDsoHandle() {
 }
 
 port::StatusOr<void*> GetCusolverDsoHandle() {
-  return GetDsoHandle("cusolver", GetCudaVersion());
+  return GetDsoHandle("cusolver", GetCudaLibVersion());
 }
 
 port::StatusOr<void*> GetCusparseDsoHandle() {
-  return GetDsoHandle("cusparse", GetCudaVersion());
+  return GetDsoHandle("cusparse", GetCudaLibVersion());
 }
 
 port::StatusOr<void*> GetCurandDsoHandle() {