Fix CUDA build

1 files changed, 8 insertions, 2 deletions

diff --git a/PKGBUILD b/PKGBUILD
index 97dc21bb57af..d04907844b28 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -23,6 +23,9 @@ makedepends=('python-setuptools')
 if $_with_r; then
   makedepends+=('r' 'r-stringi' 'r-magrittr' 'r-data.table')
 fi
+if $_with_cuda; then
+  makedepends+=('gcc11')
+fi
 arch=('x86_64')
 sha256sums=('SKIP'
             'SKIP'
@@ -55,11 +58,14 @@ build() {
     cmake_args+=('-DR_LIB=ON')
   fi
   if $_with_cuda; then
-    cmake_args+=('-DUSE_CUDA=ON')
+    cmake_args+=('-DUSE_CUDA=ON' '-DBUILD_WITH_CUDA_CUB=ON')
+    if [ -n "$_compute_ver" ]; then
+      cmake_args+=("-DGPU_COMPUTE_VER=$_compute_ver")
+    fi
   fi
 
   if $_with_cuda; then
-    CC=/usr/bin/gcc-8 CXX=/usr/bin/g++-8 cmake .. ${cmake_args[@]}
+    CC=/usr/bin/gcc-11 CXX=/usr/bin/g++-11 cmake .. ${cmake_args[@]}
   else
     cmake .. ${cmake_args[@]}
   fi