diff options
Diffstat (limited to 'PKGBUILD')
-rw-r--r-- | PKGBUILD | 29 |
1 files changed, 3 insertions, 26 deletions
@@ -5,9 +5,9 @@ pkgbase=python-pytorch-rocm pkgname=("python-pytorch-rocm" "python-pytorch-opt-rocm") _pkgname="pytorch" -pkgver=1.6.0 -_pkgver=1.6.0 -pkgrel=8 +pkgver=1.7.0rc1 +_pkgver=1.7.0-rc1 +pkgrel=1 pkgdesc="Tensors and Dynamic neural networks in Python with strong GPU acceleration" arch=('x86_64') url="https://pytorch.org" @@ -18,19 +18,9 @@ makedepends=('python' 'python-setuptools' 'python-yaml' 'python-numpy' 'cmake' ' 'git' 'magma' 'ninja' 'pkgconfig' 'doxygen') source=("${_pkgname}-${pkgver}::git+https://github.com/pytorch/pytorch.git#tag=v$_pkgver" fix_include_system.patch - use-system-libuv.patch - use-system-libuv2.patch - nccl_version.patch - find-hip.patch - "find-rccl.patch::https://patch-diff.githubusercontent.com/raw/pytorch/pytorch/pull/42072.patch" "find-hsa-runtime.patch::https://patch-diff.githubusercontent.com/raw/pytorch/pytorch/pull/45550.patch") sha256sums=('SKIP' '147bdaeac8ec46ea46382e6146878bd8f8d51e05d5bd6f930dfd8e2b520859b9' - '6f3b7a87172011de810bf1ab581245b4463ef86e5cd09bec63aeffa372e26646' - '7b65c3b209fc39f92ba58a58be6d3da40799f1922910b1171ccd9209eda1f9eb' - '1a276bd827a0c76dab908cbc6605fa4c9fc2cc2b9431b6578a41133ae27dba2b' - 'c41b5a95513d8ff533eb083eee8e9106553669ef6fe4da238729d5dbd92f969f' - 'SKIP' 'SKIP') prepare() { @@ -46,19 +36,6 @@ prepare() { # https://bugs.archlinux.org/task/64981 patch -N torch/utils/cpp_extension.py "${srcdir}"/fix_include_system.patch - # Use system libuv - patch -Np1 -i "${srcdir}"/use-system-libuv.patch - patch -Np1 -i "${srcdir}"/use-system-libuv2.patch -d third_party/tensorpipe - - # FindNCCL patch to export correct nccl version - patch -Np1 -i "${srcdir}"/nccl_version.patch - - # https://github.com/pytorch/pytorch/issues/41886 - patch -Np1 -i "${srcdir}"/find-hip.patch - - # https://github.com/pytorch/pytorch/pull/42072 - patch -Np1 -i "${srcdir}"/find-rccl.patch - # https://github.com/pytorch/pytorch/pull/45550 patch -Np1 -i "${srcdir}"/find-hsa-runtime.patch |