diff options
author | acxz | 2022-07-06 15:03:21 -0400 |
---|---|---|
committer | acxz | 2022-07-06 15:03:21 -0400 |
commit | a499ca51ef1fd07bd78fa26be2e9443e8f0bcd8b (patch) | |
tree | 6eb05a84fd1c92b3bdc87fdea7cfde8791d7f43f | |
parent | f032cf7258f5d1ed37314cc2df92ecf25b35ec91 (diff) | |
download | aur-a499ca51ef1fd07bd78fa26be2e9443e8f0bcd8b.tar.gz |
upgpkg: tensorflow-rocm 2.9.1-1
-rw-r--r-- | .SRCINFO | 12 | ||||
-rw-r--r-- | PKGBUILD | 32 |
2 files changed, 26 insertions, 18 deletions
@@ -1,7 +1,7 @@ pkgbase = tensorflow-rocm pkgdesc = Library for computation using data flow graphs for scalable machine learning - pkgver = 2.8.0 - pkgrel = 2 + pkgver = 2.9.1 + pkgrel = 1 url = https://www.tensorflow.org/ arch = x86_64 license = APACHE @@ -30,10 +30,12 @@ pkgbase = tensorflow-rocm depends = icu depends = libjpeg-turbo optdepends = tensorboard: Tensorflow visualization toolkit - source = tensorflow-rocm-2.8.0.tar.gz::https://github.com/tensorflow/tensorflow/archive/v2.8.0.tar.gz + source = tensorflow-rocm-2.9.1.tar.gz::https://github.com/tensorflow/tensorflow/archive/v2.9.1.tar.gz source = fix-c++17-compat.patch - sha512sums = 9cddb78c0392b7810e71917c3731f895e31c250822031ac7f498bf20435408c640b2fba4de439fa4a47c70dbff38b86e50fed2971df1f1916f23f9490241cfed + source = rocblas-version.patch::https://github.com/tensorflow/tensorflow/commit/9ee3b8a47ab5b5307feba0943d2f5ca46c91ceac.patch + sha512sums = 95ffbee1e50e396065c6f1802fd9668344c45c000e22da859bcd08ec217bcc0a8ff0e84661fdf511f210e8b09d7ae6d26c3fc1ddcf28b8aedf87c0fb1b8b60e4 sha512sums = f682368bb47b2b022a51aa77345dfa30f3b0d7911c56515d428b8326ee3751242f375f4e715a37bb723ef20a86916dad9871c3c81b1b58da85e1ca202bc4901e + sha512sums = SKIP pkgname = tensorflow-rocm pkgdesc = Library for computation using data flow graphs for scalable machine learning (with ROCM) @@ -123,7 +125,7 @@ pkgname = python-tensorflow-opt-rocm depends = giflib depends = icu depends = libjpeg-turbo - depends = tensorflow-rocm + depends = tensorflow-opt-rocm depends = rocm-hip-sdk depends = miopen depends = rccl @@ -14,9 +14,9 @@ pkgname=() [ "$_build_no_opt" -eq 1 ] && pkgname+=(tensorflow-rocm python-tensorflow-rocm) [ "$_build_opt" -eq 1 ] && pkgname+=(tensorflow-opt-rocm python-tensorflow-opt-rocm) -pkgver=2.8.0 -_pkgver=2.8.0 -pkgrel=2 +pkgver=2.9.1 +_pkgver=2.9.1 +pkgrel=1 pkgdesc="Library for computation using data flow graphs for scalable machine learning" url="https://www.tensorflow.org/" license=('APACHE') @@ -28,10 +28,12 @@ makedepends=('bazel' 'python-numpy' 'rocm-hip-sdk' 'miopen' 'rccl' 'git' '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 + "rocblas-version.patch::https://github.com/tensorflow/tensorflow/commit/9ee3b8a47ab5b5307feba0943d2f5ca46c91ceac.patch") -sha512sums=('9cddb78c0392b7810e71917c3731f895e31c250822031ac7f498bf20435408c640b2fba4de439fa4a47c70dbff38b86e50fed2971df1f1916f23f9490241cfed' - 'f682368bb47b2b022a51aa77345dfa30f3b0d7911c56515d428b8326ee3751242f375f4e715a37bb723ef20a86916dad9871c3c81b1b58da85e1ca202bc4901e') +sha512sums=('95ffbee1e50e396065c6f1802fd9668344c45c000e22da859bcd08ec217bcc0a8ff0e84661fdf511f210e8b09d7ae6d26c3fc1ddcf28b8aedf87c0fb1b8b60e4' + 'f682368bb47b2b022a51aa77345dfa30f3b0d7911c56515d428b8326ee3751242f375f4e715a37bb723ef20a86916dad9871c3c81b1b58da85e1ca202bc4901e' + 'SKIP') # consolidate common dependencies to prevent mishaps _common_py_depends=(python-termcolor python-astor python-gast03 python-numpy python-protobuf @@ -72,6 +74,10 @@ prepare() { # 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 + cd "${srcdir}/tensorflow-${pkgver}" + patch -Np1 -i "${srcdir}/rocblas-version.patch" + cd "${srcdir}" + cp -r tensorflow-${_pkgver} tensorflow-${_pkgver}-rocm cp -r tensorflow-${_pkgver} tensorflow-${_pkgver}-opt-rocm @@ -103,9 +109,12 @@ prepare() { export TF_IGNORE_MAX_BAZEL_VERSION=1 export TF_MKL_ROOT=/opt/intel/mkl export NCCL_INSTALL_PATH=/usr - export GCC_HOST_COMPILER_PATH=/usr/bin/gcc - export HOST_C_COMPILER=/usr/bin/gcc - export HOST_CXX_COMPILER=/usr/bin/g++ + # Does tensorflow really need the compiler overridden in 5 places? Yes. + export CC=gcc-11 + export CXX=g++-11 + export GCC_HOST_COMPILER_PATH=/usr/bin/${CC} + export HOST_C_COMPILER=/usr/bin/${CC} + export HOST_CXX_COMPILER=/usr/bin/${CXX} export TF_CUDA_CLANG=0 # Clang currently disabled because it's not compatible at the moment. export CLANG_CUDA_COMPILER_PATH=/usr/bin/clang export TF_CUDA_PATHS=/opt/cuda,/usr/lib,/usr @@ -116,9 +125,6 @@ prepare() { # 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 CC=gcc - export CXX=g++ - export BAZEL_ARGS="--config=mkl -c opt" } @@ -258,7 +264,7 @@ package_python-tensorflow-rocm() { package_python-tensorflow-opt-rocm() { pkgdesc="Library for computation using data flow graphs for scalable machine learning (with ROCM and AVX2 CPU optimizations)" - depends+=(tensorflow-rocm rocm-hip-sdk miopen rccl "${_common_py_depends[@]}") + depends+=(tensorflow-opt-rocm rocm-hip-sdk miopen rccl "${_common_py_depends[@]}") conflicts=(python-tensorflow) provides=(python-tensorflow python-tensorflow-rocm) |