Add GPU Compute Capability to fix Segmentation fault on RTX3090.

[kkongling]

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 21809c0..06ef45d 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -15,6 +15,10 @@ add_definitions (-DCMAKE_BUILD_TYPE=${CMAKE_BUILD_TYPE})

 find_package(CUDA REQUIRED)

+#foreach(ComputeCapability 60 61 70 72 75 86)
+#  set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS};-gencode;arch=compute_${ComputeCapability},code=sm_${ComputeCapability})
+#endforeach()
+
 if (CMAKE_BUILD_TYPE MATCHES Debug)
   add_definitions (-DDEBUG)
 endif (CMAKE_BUILD_TYPE MATCHES Debug)

refer to Your GPU Compute Capability

[nepluno]

Thanks for reporting the issue! Fixed in the latest commit according to your suggestion.