Date: Monday, August 22, 2022 @ 18:27:57 Author: kgizdov Revision: 1274816
upgpkg: tensorflow 2.9.1-3: cuda arch update and nccl rebuild Added: tensorflow/trunk/fix-cusolver-version.patch Modified: tensorflow/trunk/PKGBUILD ----------------------------+ PKGBUILD | 13 +++++++++---- fix-cusolver-version.patch | 12 ++++++++++++ 2 files changed, 21 insertions(+), 4 deletions(-) Modified: PKGBUILD =================================================================== --- PKGBUILD 2022-08-22 18:06:30 UTC (rev 1274815) +++ PKGBUILD 2022-08-22 18:27:57 UTC (rev 1274816) @@ -7,7 +7,7 @@ pkgname=(tensorflow tensorflow-opt tensorflow-cuda tensorflow-opt-cuda python-tensorflow python-tensorflow-opt python-tensorflow-cuda python-tensorflow-opt-cuda) pkgver=2.9.1 _pkgver=2.9.1 -pkgrel=2 +pkgrel=3 pkgdesc="Library for computation using data flow graphs for scalable machine learning" url="https://www.tensorflow.org/"; license=('APACHE') @@ -18,9 +18,11 @@ 'cython') optdepends=('tensorboard: Tensorflow visualization toolkit') source=("$pkgname-$pkgver.tar.gz::https://github.com/tensorflow/tensorflow/archive/v${_pkgver}.tar.gz"; - fix-c++17-compat.patch) + fix-c++17-compat.patch + fix-cusolver-version.patch) sha512sums=('95ffbee1e50e396065c6f1802fd9668344c45c000e22da859bcd08ec217bcc0a8ff0e84661fdf511f210e8b09d7ae6d26c3fc1ddcf28b8aedf87c0fb1b8b60e4' - 'f682368bb47b2b022a51aa77345dfa30f3b0d7911c56515d428b8326ee3751242f375f4e715a37bb723ef20a86916dad9871c3c81b1b58da85e1ca202bc4901e') + 'f682368bb47b2b022a51aa77345dfa30f3b0d7911c56515d428b8326ee3751242f375f4e715a37bb723ef20a86916dad9871c3c81b1b58da85e1ca202bc4901e' + '6f42455db1db0a5cd58ab5fe5554317e9ff648c046bb81cef9b4c61cce8380da08b681f825544b5388f02da863ff19f642efa9459691cbcf8852a21bd0dc7447') # consolidate common dependencies to prevent mishaps _common_py_depends=(python-termcolor python-astor python-gast03 python-numpy python-protobuf @@ -61,6 +63,9 @@ # thinks about which versions should be used anyway. ;) (FS#68772) sed -i -E "s/'([0-9a-z_-]+) .= [0-9].+[0-9]'/'\1'/" tensorflow-${_pkgver}/tensorflow/tools/pip_package/setup.py + # manually specify cusolver .so version + patch -Np1 -i "${srcdir}/fix-cusolver-version.patch" -d tensorflow-${_pkgver} + cp -r tensorflow-${_pkgver} tensorflow-${_pkgver}-opt cp -r tensorflow-${_pkgver} tensorflow-${_pkgver}-cuda cp -r tensorflow-${_pkgver} tensorflow-${_pkgver}-opt-cuda @@ -105,7 +110,7 @@ # https://github.com/tensorflow/tensorflow/blob/1ba2eb7b313c0c5001ee1683a3ec4fbae01105fd/third_party/gpus/cuda_configure.bzl#L411-L446 # according to the above, we should be specifying CUDA compute capabilities as 'sm_XX' or 'compute_XX' from now on # add latest PTX for future compatibility - export TF_CUDA_COMPUTE_CAPABILITIES=sm_52,sm_53,sm_60,sm_61,sm_62,sm_70,sm_72,sm_75,sm_80,sm_86,compute_86 + export TF_CUDA_COMPUTE_CAPABILITIES=sm_52,sm_53,sm_60,sm_61,sm_62,sm_70,sm_72,sm_75,sm_80,sm_86,sm_87,compute_87 export BAZEL_ARGS="--config=mkl -c opt" } Added: fix-cusolver-version.patch =================================================================== --- fix-cusolver-version.patch (rev 0) +++ fix-cusolver-version.patch 2022-08-22 18:27:57 UTC (rev 1274816) @@ -0,0 +1,12 @@ +diff --color -aur tensorflow-2.9.1-old/third_party/gpus/cuda_configure.bzl tensorflow-2.9.1-new/third_party/gpus/cuda_configure.bzl +--- tensorflow-2.9.1-old/third_party/gpus/cuda_configure.bzl 2022-08-22 14:03:54.439731148 +0300 ++++ tensorflow-2.9.1-new/third_party/gpus/cuda_configure.bzl 2022-08-22 14:07:09.479726331 +0300 +@@ -710,7 +716,7 @@ + cuda_version_major = cuda_major, + cudart_version = cudart_version, + cublas_version = cublas_version, +- cusolver_version = cusolver_version, ++ cusolver_version = "11", + curand_version = curand_version, + cufft_version = cufft_version, + cusparse_version = cusparse_version,
