From baffa738c8a72149e8279edf099f7e832c0d5790 Mon Sep 17 00:00:00 2001
From: Simon Rit <simon.rit@creatis.insa-lyon.fr>
Date: Thu, 14 Oct 2021 15:00:45 +0200
Subject: [PATCH] COMP: use 5.2 as the default Cuda architecture

---
 itk-module-init.cmake | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/itk-module-init.cmake b/itk-module-init.cmake
index d5de5d140..9a7e0785d 100644
--- a/itk-module-init.cmake
+++ b/itk-module-init.cmake
@@ -3,6 +3,16 @@
 #
 list(APPEND CMAKE_MODULE_PATH ${CMAKE_CURRENT_LIST_DIR}/cmake)
 
+# Set default Cuda architecture if not provided. The first case allows for
+# backward compatibility with cmake versions before 3.20 which did not handle
+# CUDAARCHS environment variable.
+if(NOT DEFINED CMAKE_CUDA_ARCHITECTURES)
+  if(DEFINED ENV{CUDAARCHS})
+    set(CMAKE_CUDA_ARCHITECTURES "$ENV{CUDAARCHS}" CACHE STRING "CUDA architectures")
+  else()
+    set(CMAKE_CUDA_ARCHITECTURES "52" CACHE STRING "CUDA architectures")
+  endif()
+endif()
 include(CheckLanguage)
 check_language(CUDA)