Expand arch support in CMakelists.txt

CMakelists.txt

@@ -22,8 +22,8 @@ set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} /fp:fast")
 if( MAKE_CUDA_BUILD ) 
 
     # Appending -real to target archs to discard unneeded PTX code
-    # No support for compute less than 7.0 in latest version
-    set(COMPUTE_CAPABILITY_NOCUBLAST 70-real 72-real)
+    set(COMPUTE_CAPABILITY_NOCUBLAST 50-real 52-real 60-real 61-real 70-real 72-real)
+    set(COMPUTE_CAPABILITY_KEPLER 35-real 37-real)
 
     # Later versions of CUDA support the new architectures   forgoing -real on latest arch to support future archs
     set(CC_CUDA10x 75)
@@ -51,7 +51,7 @@ if( MAKE_CUDA_BUILD )
     if(${CUDA_TARGET_ARCH_FEATURE_LEVEL} STREQUAL "11.0")
         set(CMAKE_CUDA_ARCHITECTURES ${CC_CUDA110})
         if(NO_CUBLASLT)
-            list(APPEND CMAKE_CUDA_ARCHITECTURES ${COMPUTE_CAPABILITY_NOCUBLAST})
+            list(APPEND CMAKE_CUDA_ARCHITECTURES ${COMPUTE_CAPABILITY_NOCUBLAST} ${COMPUTE_CAPABILITY_KEPLER})
         endif()
 
     elseif(${CUDA_VERSION_TARGET_FEATURE_MAJOR} STREQUAL "11")
@@ -60,7 +60,7 @@ if( MAKE_CUDA_BUILD )
             set(CMAKE_CUDA_ARCHITECTURES ${CC_CUDA11x})
         endif()
         if(NO_CUBLASLT)
-            list(APPEND CMAKE_CUDA_ARCHITECTURES ${COMPUTE_CAPABILITY_NOCUBLAST})
+            list(APPEND CMAKE_CUDA_ARCHITECTURES ${COMPUTE_CAPABILITY_NOCUBLAST} ${COMPUTE_CAPABILITY_KEPLER})
         endif()
     elseif(${CUDA_VERSION_TARGET_FEATURE_MAJOR} STREQUAL "12")
         set(CMAKE_CUDA_ARCHITECTURES ${CC_CUDA118})
@@ -69,7 +69,7 @@ if( MAKE_CUDA_BUILD )
         endif()
     else()
         set(CMAKE_CUDA_ARCHITECTURES ${CC_CUDA10x})
-        list(APPEND CMAKE_CUDA_ARCHITECTURES ${COMPUTE_CAPABILITY_NOCUBLAST})
+        list(APPEND CMAKE_CUDA_ARCHITECTURES ${COMPUTE_CAPABILITY_NOCUBLAST} ${COMPUTE_CAPABILITY_KEPLER})
         set(NO_CUBLASLT ON)
     endif()