summaryrefslogtreecommitdiffstats
path: root/build/linux/configure.ac
diff options
context:
space:
mode:
authorWillem Jan Palenstijn <Willem.Jan.Palenstijn@cwi.nl>2014-10-02 15:19:47 +0200
committerWillem Jan Palenstijn <Willem.Jan.Palenstijn@cwi.nl>2014-10-02 15:22:22 +0200
commit52d34e926ee5a42d875dede9303ae898548fdc85 (patch)
treeff701c46c8354750d9320963720cdfbd5837220c /build/linux/configure.ac
parent5ecd2b05bfcb6ffb7216f9ae0505aa451306e598 (diff)
downloadastra-52d34e926ee5a42d875dede9303ae898548fdc85.tar.gz
astra-52d34e926ee5a42d875dede9303ae898548fdc85.tar.bz2
astra-52d34e926ee5a42d875dede9303ae898548fdc85.tar.xz
astra-52d34e926ee5a42d875dede9303ae898548fdc85.zip
Add --with-cuda-compute option to configure
Diffstat (limited to 'build/linux/configure.ac')
-rw-r--r--build/linux/configure.ac8
1 files changed, 6 insertions, 2 deletions
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