summaryrefslogtreecommitdiffstats
path: root/build
diff options
context:
space:
mode:
Diffstat (limited to 'build')
-rw-r--r--build/linux/configure.ac16
1 files changed, 14 insertions, 2 deletions
diff --git a/build/linux/configure.ac b/build/linux/configure.ac
index 37fa223..5ff4eb6 100644
--- a/build/linux/configure.ac
+++ b/build/linux/configure.ac
@@ -133,9 +133,21 @@ fi
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 21 30 32 35 37 50 52 53 60 61 62 70 72 75
+ dnl 10 11 12 13 20 21 30 32 35 37 50 52 53 60 61 62 70 72 75 80 86
+ dnl
+ dnl Build cubin for range of platforms, and ptx for newest for forward compat.
+ dnl Build cubin x.0 for older platforms, and all x.y for "recent" ones.
+ dnl
+ dnl Include 3.5 since CUDA 11.x doesn't support 3.0, but does support 3.5.
+ dnl (All other CUDA toolkits from 7-10 start support at x.0 versions.)
+ dnl Skip 7.2 cubin since that seems to be Jetson-only (and can use 7.0 cubin).
+ dnl
+ dnl From CUDA Programming Guide:
+ dnl "Binary compatibility is guaranteed from one minor revision to the next one, but not from one minor revision to the previous one or across major revisions."
+ dnl
+
if test x"$with_cuda_compute" = x; then
- with_cuda_compute="20,30,35,50,60,70,75"
+ with_cuda_compute="20,30,35,50,60,70,75,80,86"
fi
ASTRA_FIND_NVCC_ARCHS([$with_cuda_compute],NVCCFLAGS_EXTRA,NVCCARCHS)
AC_MSG_RESULT([$NVCCARCHS])