blob: 825963c5ebb3588c07b44ab2f62481d20f29cf9d (
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
|
# Maintainer: piernov <piernov@piernov.org>
pkgname=python-pytorch3d
pkgver=0.7.6
pkgrel=1
pkgdesc="Provides efficient, reusable components for 3D Computer Vision research with PyTorch."
arch=('x86_64')
url="https://pytorch3d.org/"
license=('BSD')
depends=('python-fvcore' 'python-iopath')
makedepends=('python-setuptools')
source=("$pkgname-$pkgver.tar.gz::https://github.com/facebookresearch/pytorch3d/archive/v$pkgver.tar.gz"
'setup-py_extensions_relative_paths.patch')
md5sums=('66bece60c411256ea463d5f632a747ab'
'506f83849dd3afab00501fc2442473fb')
build() {
cd "$srcdir"/pytorch3d-$pkgver
# Avoid references to $srcdir in SOURCES.txt and native Python extension
patch -p1 < "${srcdir}/setup-py_extensions_relative_paths.patch"
CFLAGS+=" -ffile-prefix-map=$PWD/="
CXXFLAGS+=" -ffile-prefix-map=$PWD/="
NVCC_FLAGS+="-Xcompiler -ffile-prefix-map=$PWD/="
export NVCC_FLAGS
python setup.py build
}
package() {
cd "$srcdir"/pytorch3d-$pkgver
python setup.py install --root="$pkgdir"/ --optimize=1 --skip-build
}
|