blob: 3668290fbac3f433cf9287ee627643d7ff6ec34f (
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
|
# Maintainer: fanyujun <dlutfyj@outlook.com>
# Contributor:
pkgname="python-thop"
_pkgname="pytorch-OpCounter"
epoch=1
pkgver=0.1.1.post2209072238
pkgrel=1
pkgdesc='Count the MACs / FLOPs of your PyTorch model.'
arch=(any)
url="https://github.com/Lyken17/pytorch-OpCounter"
license=(MIT)
depends=(python python-pytorch)
conflicts=()
makedepends=(python-build python-installer python-setuptools python-wheel)
source=("git+$url#commit=43c064a")
sha256sums=('SKIP')
build() {
cd "${srcdir}/${_pkgname}"
sed -i '21d' setup.py
python -m build --wheel --skip-dependency-check --no-isolation
}
package(){
depends+=()
cd "${srcdir}/${_pkgname}"
PYTHONPYCACHEPREFIX="${PWD}/.cache/cpython/" python -m installer --destdir="${pkgdir}" dist/*.whl
install -Dm 644 LICENSE -t "$pkgdir/usr/share/licenses/$pkgname/"
}
|