Update to version 0.8.2_1.8.1

2 files changed, 5 insertions, 4 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 539b5f8ed00b..1790af304193 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,6 +1,6 @@
 pkgbase = caffe2
 	pkgdesc = A new lightweight, modular, and scalable deep learning framework
-	pkgver = 0.8.2_1.8.0
+	pkgver = 0.8.2_1.8.1
 	pkgrel = 1
 	url = https://caffe2.ai/
 	arch = x86_64
@@ -46,7 +46,7 @@ pkgbase = caffe2
 	optdepends = python-lmdb
 	conflicts = python-pytorch
 	options = !emptydirs
-	source = git+https://github.com/pytorch/pytorch.git#tag=v1.8.0
+	source = git+https://github.com/pytorch/pytorch.git#tag=v1.8.1
 	source = git+https://github.com/pybind/pybind11.git
 	source = cub-nvlabs::git+https://github.com/NVlabs/cub.git
 	source = git+https://github.com/eigenteam/eigen-git-mirror.git
diff --git a/PKGBUILD b/PKGBUILD
index e100b0d52894..844efdb5ea6f 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -6,7 +6,7 @@ _cuda_archs='5.2;5.3;6.0;6.1;6.2;7.0;7.0+PTX;7.2;7.2+PTX;7.5;7.5+PTX;8.0;8.0+PTX
 
 pkgbase=caffe2
 pkgname=('caffe2' 'caffe2-cuda')
-_pytorchver=1.8.0
+_pytorchver=1.8.1
 pkgver="0.8.2_${_pytorchver}"
 pkgrel=1
 pkgdesc='A new lightweight, modular, and scalable deep learning framework'
@@ -129,7 +129,8 @@ prepare() {
                            'foxi'
                            'tbb'
                            'fmt'
-                           'tensorpipe')
+                           'tensorpipe'
+                           'kineto')
     
     git -C pytorch submodule init