blob: ba560eb4a622543b4702ba7152998c46e8af0002 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
pkgbase = python-pytorch-rocm
pkgdesc = Tensors and Dynamic neural networks in Python with strong GPU acceleration
pkgver = 1.7.0rc2
pkgrel = 1
url = https://pytorch.org
arch = x86_64
license = BSD
makedepends = python
makedepends = python-setuptools
makedepends = python-yaml
makedepends = python-numpy
makedepends = cmake
makedepends = rocm
makedepends = rocm-libs
makedepends = miopen
makedepends = git
makedepends = magma
makedepends = ninja
makedepends = pkgconfig
makedepends = doxygen
depends = google-glog
depends = gflags
depends = opencv
depends = openmp
depends = rccl
depends = pybind11
depends = python
depends = python-yaml
depends = libuv
depends = python-numpy
depends = protobuf
depends = ffmpeg
depends = python-future
depends = qt5-base
depends = onednn
depends = intel-mkl
source = pytorch-1.7.0rc2::git+https://github.com/pytorch/pytorch.git#tag=v1.7.0-rc2
source = fix_include_system.patch
source = find-hsa-runtime.patch::https://patch-diff.githubusercontent.com/raw/pytorch/pytorch/pull/45550.patch
sha256sums = SKIP
sha256sums = 147bdaeac8ec46ea46382e6146878bd8f8d51e05d5bd6f930dfd8e2b520859b9
sha256sums = SKIP
pkgname = python-pytorch-rocm
pkgdesc = Tensors and Dynamic neural networks in Python with strong GPU acceleration (with ROCM)
depends = google-glog
depends = gflags
depends = opencv
depends = openmp
depends = rccl
depends = pybind11
depends = python
depends = python-yaml
depends = libuv
depends = python-numpy
depends = protobuf
depends = ffmpeg
depends = python-future
depends = qt5-base
depends = onednn
depends = intel-mkl
depends = rocm
depends = magma
provides = python-pytorch
conflicts = python-pytorch
pkgname = python-pytorch-opt-rocm
pkgdesc = Tensors and Dynamic neural networks in Python with strong GPU acceleration (with ROCM and CPU optimizations)
depends = google-glog
depends = gflags
depends = opencv
depends = openmp
depends = rccl
depends = pybind11
depends = python
depends = python-yaml
depends = libuv
depends = python-numpy
depends = protobuf
depends = ffmpeg
depends = python-future
depends = qt5-base
depends = onednn
depends = intel-mkl
depends = rocm
depends = magma
provides = python-pytorch
provides = python-pytorch-rocm
conflicts = python-pytorch
|