[lilac] updated to 0.7.0-4

2 files changed, 6 insertions, 4 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 75912f0d15c3..05ba9b8be14e 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,7 +1,7 @@
 pkgbase = python-torchvision
 	pkgdesc = Datasets, transforms, and models specific to computer vision
 	pkgver = 0.7.0
-	pkgrel = 2
+	pkgrel = 4
 	url = https://github.com/pytorch/vision
 	arch = x86_64
 	license = BSD
@@ -21,6 +21,7 @@ pkgbase = python-torchvision
 	depends = python-tqdm
 	optdepends = ffmpeg: video reader backend (the recommended one with better performance)
 	optdepends = python-av: video reader backend (the default one)
+	optdepends = python-pycocotools: support for MS-COCO dataset
 	source = vision-0.7.0.tar.gz::https://github.com/pytorch/vision/archive/v0.7.0.tar.gz
 	sha512sums = f22c8fd67f4efafa4a934ed5014e1ad661d0d0a800d4c8ae0e6b85f4860a5dc131c513428028949c9130486ba24ecf8a9f6b9881efc68a0c2158110924d05e81
 
diff --git a/PKGBUILD b/PKGBUILD
index b3ee63cf3595..2ee1e5549f3c 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -7,7 +7,7 @@
 pkgname=('python-torchvision' 'python-torchvision-cuda')
 _pkgname=vision
 pkgver=0.7.0
-pkgrel=2
+pkgrel=4
 pkgdesc='Datasets, transforms, and models specific to computer vision'
 arch=('x86_64')
 url='https://github.com/pytorch/vision'
@@ -22,6 +22,7 @@ depends=(
 optdepends=(
   'ffmpeg: video reader backend (the recommended one with better performance)'
   'python-av: video reader backend (the default one)'
+  'python-pycocotools: support for MS-COCO dataset'
 )
 makedepends=(
   'cuda'
@@ -53,7 +54,7 @@ build() {
   
   cd "${srcdir}/${_pkgname}-cuda-${pkgver}"
   FORCE_CUDA=1 \
-  TORCH_CUDA_ARCH_LIST="5.2;5.3;6.0;6.0+PTX;6.1;6.1+PTX;6.2;6.2+PTX;7.0;7.0+PTX;7.2;7.2+PTX;7.5;7.5+PTX;8.0;8.0+PTX" \
+  TORCH_CUDA_ARCH_LIST="5.2;5.3;6.0;6.0+PTX;6.1;6.1+PTX;6.2;6.2+PTX;7.0;7.0+PTX;7.2;7.2+PTX;7.5;7.5+PTX;8.0" \
   python setup.py build
 }
 
@@ -81,7 +82,7 @@ package_python-torchvision-cuda() {
   
   cd "${srcdir}/${_pkgname}-cuda-${pkgver}"
   FORCE_CUDA=1 \
-  TORCH_CUDA_ARCH_LIST="5.2;5.3;6.0;6.0+PTX;6.1;6.1+PTX;6.2;6.2+PTX;7.0;7.0+PTX;7.2;7.2+PTX;7.5;7.5+PTX;8.0;8.0+PTX" \
+  TORCH_CUDA_ARCH_LIST="5.2;5.3;6.0;6.0+PTX;6.1;6.1+PTX;6.2;6.2+PTX;7.0;7.0+PTX;7.2;7.2+PTX;7.5;7.5+PTX;8.0" \
   python setup.py install --root="${pkgdir}" --optimize=1 --skip-build
   install -Dm644 LICENSE -t "${pkgdir}/usr/share/licenses/${pkgname}"
 }