From 52d34e926ee5a42d875dede9303ae898548fdc85 Mon Sep 17 00:00:00 2001
From: Willem Jan Palenstijn <Willem.Jan.Palenstijn@cwi.nl>
Date: Thu, 2 Oct 2014 15:19:47 +0200
Subject: Add --with-cuda-compute option to configure

---
 build/linux/acinclude.m4 | 6 +++++-
 build/linux/configure.ac | 8 ++++++--
 2 files changed, 11 insertions(+), 3 deletions(-)

(limited to 'build')

diff --git a/build/linux/acinclude.m4 b/build/linux/acinclude.m4
index 313e849..5022e5b 100644
--- a/build/linux/acinclude.m4
+++ b/build/linux/acinclude.m4
@@ -74,7 +74,7 @@ rm -f conftest.cu conftest.o conftest.nvcc.out
 ])
 
 dnl ASTRA_FIND_NVCC_ARCHS(archs-to-try,cppflags-to-extend,output-list)
-dnl Architectures should be of the form 10, 20, 30, 35,
+dnl Architectures should be of the form 10,20,30,35,
 dnl and should be in order. The last accepted one will be used for PTX output.
 dnl All accepted ones will be used for cubin output.
 AC_DEFUN([ASTRA_FIND_NVCC_ARCHS],[
@@ -88,7 +88,10 @@ _ACEOF
 NVCC_lastarch="none"
 NVCC_extra=""
 NVCC_list=""
+astra_save_IFS=$IFS
+IFS=,
 for arch in $1; do
+  IFS=$astra_save_IFS
   NVCC_opt="-gencode=arch=compute_$arch,code=sm_$arch"
   $NVCC -c -o conftest.o conftest.cu $$2 $NVCC_opt >conftest.nvcc.out 2>&1 && {
     NVCC_lastarch=$arch
@@ -96,6 +99,7 @@ for arch in $1; do
     NVCC_list="${NVCC_list:+$NVCC_list, }$arch"
   }
 done
+IFS=$astra_save_IFS
 if test $NVCC_lastarch != none; then
   NVCC_extra="$NVCC_extra -gencode=arch=compute_${NVCC_lastarch},code=compute_${NVCC_lastarch}"
   $3="$NVCC_list"
diff --git a/build/linux/configure.ac b/build/linux/configure.ac
index 5fe6863..18c2fb9 100644
--- a/build/linux/configure.ac
+++ b/build/linux/configure.ac
@@ -121,11 +121,15 @@ if test x"$HAVECUDA" = xyes; then
 fi
 AC_MSG_RESULT($HAVECUDA)
 
+AC_ARG_WITH(cuda_compute, [[  --with-cuda-compute=archs  comma separated list of CUDA compute models (optional)]],,)
 if test x"$HAVECUDA" = xyes; then
   AC_MSG_CHECKING([for nvcc archs])
   dnl 10 11 12 13 20 30 32 35 37 50
-  ASTRA_FIND_NVCC_ARCHS([10 12 20 30 35 50],NVCCFLAGS,NVCCARCHS)
-  AC_MSG_RESULT($NVCCARCHS)
+  if test x"$with_cuda_compute" = x; then
+    with_cuda_compute="10,12,20,30,35,50"
+  fi
+  ASTRA_FIND_NVCC_ARCHS([$with_cuda_compute],NVCCFLAGS,NVCCARCHS)
+  AC_MSG_RESULT([$NVCCARCHS])
 fi
 
 
-- 
cgit v1.2.3