summarylogtreecommitdiffstats
diff options
context:
space:
mode:
authoracxz2022-02-14 10:30:44 -0500
committeracxz2022-02-14 10:30:44 -0500
commitd2ce0a34e9e404565d1d9ac2e55e26eebe0bbe5b (patch)
tree41f9ab354e5f1d668fa9534c0ed7f585e309c451
parenta5debecd795eb4104ea2311d9f173247bd103a01 (diff)
downloadaur-d2ce0a34e9e404565d1d9ac2e55e26eebe0bbe5b.tar.gz
fix rocm dep
-rw-r--r--.SRCINFO8
-rw-r--r--PKGBUILD8
2 files changed, 7 insertions, 9 deletions
diff --git a/.SRCINFO b/.SRCINFO
index dd1a9009f42c..c0df132080b3 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,7 +1,7 @@
pkgbase = python-pytorch-rocm
pkgdesc = Tensors and Dynamic neural networks in Python with strong GPU acceleration
pkgver = 1.10.2
- pkgrel = 1
+ pkgrel = 2
url = https://pytorch.org
arch = x86_64
license = BSD
@@ -10,8 +10,7 @@ pkgbase = python-pytorch-rocm
makedepends = python-yaml
makedepends = python-numpy
makedepends = cmake
- makedepends = rocm
- makedepends = rocm-libs
+ makedepends = rocm-hip-sdk
makedepends = miopen
makedepends = git
makedepends = ninja
@@ -146,8 +145,7 @@ pkgname = python-pytorch-rocm
depends = onednn
depends = intel-mkl
depends = python-typing_extensions
- depends = rocm
- depends = rocm-libs
+ depends = rocm-hip-sdk
depends = miopen
provides = python-pytorch
conflicts = python-pytorch
diff --git a/PKGBUILD b/PKGBUILD
index 83d3f6f64ffe..f65de9544e77 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -6,7 +6,7 @@ pkgname=python-pytorch-rocm
_pkgname="pytorch"
pkgver=1.10.2
_pkgver=1.10.2
-pkgrel=1
+pkgrel=2
_pkgdesc="Tensors and Dynamic neural networks in Python with strong GPU acceleration"
pkgdesc="${_pkgdesc}"
arch=('x86_64')
@@ -15,8 +15,8 @@ license=('BSD')
depends=('google-glog' 'gflags' 'opencv' 'openmp' 'rccl' 'pybind11' 'python' 'python-yaml' 'libuv'
'python-numpy' 'protobuf' 'ffmpeg' 'python-future' 'qt5-base' 'onednn' 'intel-mkl'
'python-typing_extensions')
-makedepends=('python' 'python-setuptools' 'python-yaml' 'python-numpy' 'cmake' 'rocm'
- 'rocm-libs' 'miopen' 'git' 'ninja' 'pkgconfig' 'doxygen')
+makedepends=('python' 'python-setuptools' 'python-yaml' 'python-numpy' 'cmake' 'rocm-hip-sdk'
+ 'miopen' 'git' 'ninja' 'pkgconfig' 'doxygen')
source=("${_pkgname}-${pkgver}::git+https://github.com/pytorch/pytorch.git#tag=v$_pkgver"
# generated using parse-submodules
"${pkgname}-ios-cmake::git+https://github.com/Yangqing/ios-cmake.git"
@@ -286,7 +286,7 @@ _package() {
package_python-pytorch-rocm() {
pkgdesc="${_pkgdesc} (with ROCM and AVX2 CPU optimizations)"
- depends+=(rocm rocm-libs miopen)
+ depends+=(rocm-hip-sdk miopen)
replaces=(python-pytorch-opt-rocm)
conflicts=(python-pytorch)
provides=(python-pytorch)