Add a patch for Pascal GPU

2 files changed, 7 insertions, 5 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 6f4732afad72..f1a4ab3dd4af 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,16 +1,16 @@
 # Generated by mksrcinfo v8
-# Sun Oct 16 12:17:02 UTC 2016
+# Tue Nov  1 13:25:57 UTC 2016
 pkgbase = torch7-warp-ctc
 	pkgdesc = A fast parallel implementation of CTC, on both CPU and GPU.
 	pkgver = r56.5bfb46e
-	pkgrel = 1
+	pkgrel = 2
 	url = https://github.com/baidu-research/warp-ctc
 	arch = x86_64
 	arch = i686
 	license = APACHE
 	makedepends = cmake
 	makedepends = git
-	makedepends = gcc5
+	makedepends = torch7-cutorch-git
 	depends = torch7-git>=r819
 	optdepends = torch7-cutorch-git: For GPU support
 	source = torch7-warp-ctc::git+https://github.com/baidu-research/warp-ctc
diff --git a/PKGBUILD b/PKGBUILD
index 83d20a7595ed..62654c077434 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -2,8 +2,8 @@
 pkgdesc='A fast parallel implementation of CTC, on both CPU and GPU.'
 pkgname='torch7-warp-ctc'
 pkgver=r56.5bfb46e
-pkgrel=1
-makedepends=('cmake' 'git' 'gcc5')
+pkgrel=2
+makedepends=('cmake' 'git' 'torch7-cutorch-git')
 depends=('torch7-git>=r819')
 optdepends=('torch7-cutorch-git:	For GPU support')
 arch=('x86_64' 'i686')
@@ -24,6 +24,8 @@ pkgver () {
 build () {
 	cd "${pkgname}"
 	sed 's/luajit/luajit-5.1/g' -i CMakeLists.txt
+	# Uncomment this line if you are using a Pascal GPU
+	# sed -e 's/compute_[0-9]\{2\}/compute_61/g' -e 's/sm_[0-9]\{2\}/sm_61/g' -i CMakeLists.txt
 	cmake . -DCMAKE_INSTALL_PREFIX=/usr -DCMAKE_C_COMPILER=gcc-5 -DCMAKE_BUILD_TYPE=Release
 	make
 }