[arch-commits] Commit in tensorflow/trunk (PKGBUILD cuda_config.patch)
Konstantin Gizdov
kgizdov at archlinux.org
Tue Apr 23 16:42:19 UTC 2019
Date: Tuesday, April 23, 2019 @ 16:42:18
Author: kgizdov
Revision: 453158
correct CUDA SONAMEs at runtime
Modified:
tensorflow/trunk/PKGBUILD
tensorflow/trunk/cuda_config.patch
-------------------+
PKGBUILD | 2 +-
cuda_config.patch | 32 +++++++++++++++++++++++++++++++-
2 files changed, 32 insertions(+), 2 deletions(-)
Modified: PKGBUILD
===================================================================
--- PKGBUILD 2019-04-23 14:51:17 UTC (rev 453157)
+++ PKGBUILD 2019-04-23 16:42:18 UTC (rev 453158)
@@ -24,7 +24,7 @@
sha512sums=('1e5c020a82b0b502c9c6fc9439905e19e8f1fb07e32740a5e2c70f4a3565c8c87053ccdcad360e95550dad648154e399ca8e352fb97a1788acfec3b87bf323bb'
'770afd62612abac4fcef6919f75d1ac3519030d917ce8b277de4f88b76f5449c2ad7af81fbbcd17f5e799d637f188de200f078211f13809373d8f0378bb01754'
'5900002170aab40144453d3d0d1b097475d0cbc48c5901147214e38ed7d8634ea9207fc187b44129c878b0355676a531cb4807d4f2124f6110a20cbb61172c18'
- '0e488349ecc8e8ecf07a435196bb2c29bcbe9f139d39ef386a3d30a4162ec3496861c303a9c88a45e70c0fa57a0282c286453d57ffe289c0015b45875f8fe42c')
+ 'a4cf1779b41fdb49a09889f190db99bbd2c903f5addde0e9bfad4c2c0e02e599d9742e64172bae322f3a3e12aba79f2c92437492355124b3687e11a69c363faa')
get_pyver () {
python -c 'import sys; print(str(sys.version_info[0]) + "." + str(sys.version_info[1]))'
Modified: cuda_config.patch
===================================================================
--- cuda_config.patch 2019-04-23 14:51:17 UTC (rev 453157)
+++ cuda_config.patch 2019-04-23 16:42:18 UTC (rev 453158)
@@ -1,5 +1,35 @@
+diff --git a/tensorflow/stream_executor/dso_loader.cc b/tensorflow/stream_executor/dso_loader.cc
+index 6dda5d6315..4df1d0dcfc 100644
+--- a/tensorflow/stream_executor/dso_loader.cc
++++ b/tensorflow/stream_executor/dso_loader.cc
+@@ -46,7 +46,7 @@ string GetCudnnVersion() { return TF_CUDNN_VERSION; }
+
+ /* static */ port::Status DsoLoader::GetCublasDsoHandle(void** dso_handle) {
+ return GetDsoHandle(FindDsoPath(port::Env::Default()->FormatLibraryFileName(
+- "cublas", GetCudaVersion()),
++ "cublas", "10"),
+ GetCudaLibraryDirPath()),
+ dso_handle);
+ }
+@@ -63,14 +63,14 @@ string GetCudnnVersion() { return TF_CUDNN_VERSION; }
+
+ /* static */ port::Status DsoLoader::GetCufftDsoHandle(void** dso_handle) {
+ return GetDsoHandle(FindDsoPath(port::Env::Default()->FormatLibraryFileName(
+- "cufft", GetCudaVersion()),
++ "cufft", "10"),
+ GetCudaLibraryDirPath()),
+ dso_handle);
+ }
+
+ /* static */ port::Status DsoLoader::GetCurandDsoHandle(void** dso_handle) {
+ return GetDsoHandle(FindDsoPath(port::Env::Default()->FormatLibraryFileName(
+- "curand", GetCudaVersion()),
++ "curand", "10"),
+ GetCudaLibraryDirPath()),
+ dso_handle);
+ }
diff --git a/third_party/gpus/cuda_configure.bzl b/third_party/gpus/cuda_configure.bzl
-index 8aa5b89cdd..843e1c87bc 100644
+index 8aa5b89cdd..7ed4bb8b45 100644
--- a/third_party/gpus/cuda_configure.bzl
+++ b/third_party/gpus/cuda_configure.bzl
@@ -842,7 +842,7 @@ def _find_libs(repository_ctx, cuda_config):
More information about the arch-commits
mailing list