diff --git a/CMakeLists.txt b/CMakeLists.txt index 2437aed419b865288e218e864b2bad3a49000d4c..b475760cb44e244d279c4f6e3119de99b87dfe86 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -35,4 +35,8 @@ endif(INCLUDE_CUDA) set(SOURCES ${SOURCES_CU} ${SOURCES_CXX}) set(HEADERS ${HEADERS_CU} ${HEADERS_CXX}) -add_library(memproc STATIC ${HEADERS} ${SOURCES}) \ No newline at end of file +add_library(memproc STATIC ${HEADERS} ${SOURCES}) + +if(INCLUDE_CUDA) + set_target_properties(memproc PROPERTIES CUDA_ARCHITECTURES "60;80") +endif(INCLUDE_CUDA) \ No newline at end of file