aboutsummarylogtreecommitdiffstats
path: root/PKGBUILD
diff options
context:
space:
mode:
authorFelix Erkinger2022-08-13 22:49:32 +0200
committerAzusa Nakano2022-09-15 06:26:52 +1000
commit9328aade6ed48f8df33a4637636926c21e343c41 (patch)
tree4f726014b614eba9e98dcb56fc2e514d8ef7b855 /PKGBUILD
parent81ecf6359a67a80b3ec8c85f5e88fbee5b08c6f2 (diff)
downloadaur-9328aade6ed48f8df33a4637636926c21e343c41.tar.gz
change upstream to ROCmSoftwarePlatform/tensorflow-upstream
+ update to 2.9.2 + use branch+commit from docker.io/rocm/tensorflow@lastest at 2022-08-03
Diffstat (limited to 'PKGBUILD')
-rw-r--r--PKGBUILD9
1 files changed, 5 insertions, 4 deletions
diff --git a/PKGBUILD b/PKGBUILD
index 578ae51a07c0..9aa7b88a136f 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -3,6 +3,7 @@
# Contributor: Konstantin Gizdov (kgizdov) <arch@kge.pw>
# Contributor: Adria Arrufat (archdria) <adria.arrufat+AUR@protonmail.ch>
# Contributor: Thibault Lorrain (fredszaq) <fredszaq@gmail.com>
+# Contributor: wuxxin <wuxxin@gmail.com>
pkgbase=tensorflow-rocm
@@ -69,11 +70,11 @@ check_dir() {
prepare() {
# Allow any bazel version
- echo "*" > tensorflow-${_pkgver}/.bazelversion
+ echo "*" > $_srcname/.bazelversion
# Get rid of hardcoded versions. Not like we ever cared about what upstream
# 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
+ sed -i -E "s/'([0-9a-z_-]+) .= [0-9].+[0-9]'/'\1'/" $_srcname/tensorflow/tools/pip_package/setup.py
# manually specify cusolver .so version
patch -Np1 -i "${srcdir}/fix-cusolver-version.patch" -d tensorflow-${_pkgver}
@@ -87,8 +88,8 @@ prepare() {
cd "${srcdir}"
- cp -r tensorflow-${_pkgver} tensorflow-${_pkgver}-rocm
- cp -r tensorflow-${_pkgver} tensorflow-${_pkgver}-opt-rocm
+ cp -r $_srcname tensorflow-${_pkgver}-rocm
+ cp -r $_srcname tensorflow-${_pkgver}-opt-rocm
# These environment variables influence the behavior of the configure call below.
export PYTHON_BIN_PATH=/usr/bin/python