added patch

2 files changed, 9 insertions, 2 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 2dd93e838406..54beae43faa0 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,7 +1,7 @@
 pkgbase = torch7-cunn-git
 	pkgdesc = CUDA backend for Torch7 Neural Network Package
 	pkgver = r819.1ae6aa0
-	pkgrel = 4
+	pkgrel = 5
 	url = https://github.com/torch/cunn
 	arch = x86_64
 	arch = i686
diff --git a/PKGBUILD b/PKGBUILD
index 5f8ce9ec72fb..2ebedbd3e478 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -2,7 +2,7 @@
 pkgdesc='CUDA backend for Torch7 Neural Network Package'
 pkgname='torch7-cunn-git'
 pkgver=r819.1ae6aa0
-pkgrel=4
+pkgrel=5
 makedepends=('cmake' 'git')
 depends=('torch7-git>=r819' 'cuda' 'torch7-nn-git' 'torch7-cutorch-git')
 conflicts=('torch7-cunn')
@@ -22,6 +22,13 @@ pkgver () {
 	)
 }
 
+prepare() {
+	cd "${pkgname}"
+
+	# https://stackoverflow.com/questions/46345811/cuda-9-shfl-vs-shfl-sync
+	sed -e 's/__shfl(/__shfl_sync(0xFFFFFFFF, /g' -e 's/__any(/__any_sync(0xFFFFFFFF, /g' -i lib/THCUNN/LookupTable.cu
+}
+
 build () {
 	CFLAGS="${CFLAGS/-fno-plt/}"
 	CXXFLAGS="${CFLAGS/-fno-plt/}"