From 1d67a584799e2faa6b1bc52727420669857e35cb Mon Sep 17 00:00:00 2001
From: liulixinkerry <kerryliu1997@gmail.com>
Date: Fri, 2 Jun 2023 17:06:09 +0800
Subject: [PATCH] quick fix to support more CUDA ARCH

---
 CMakeLists.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 9f324cc..01cf612 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -76,7 +76,7 @@ message(STATUS TORCH_ENABLE_CUDA=${TORCH_ENABLE_CUDA})
 # set cuda arch and nvcc flags
 if (CUDA_FOUND)
   if (NOT CUDA_ARCH_LIST)
-    set(CUDA_ARCH_LIST 8.0 8.6)
+    set(CUDA_ARCH_LIST 6.0 6.1 7.0 7.5 8.0 8.6)
   endif(NOT CUDA_ARCH_LIST)
   # for cuda_add_library
   cuda_select_nvcc_arch_flags(CUDA_ARCH_FLAGS ${CUDA_ARCH_LIST})