diff --git a/CMakeLists.txt b/CMakeLists.txt index 400a577a..3b331d4a 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -397,6 +397,7 @@ endif() set (CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${MARCH}") + # Write out megbrain_build_config.h # It defines macros needed by both megbrain and dnn configure_file(src/megbrain_build_config.h.in ${CMAKE_CURRENT_BINARY_DIR}/genfiles/megbrain_build_config.h) diff --git a/dnn/src/CMakeLists.txt b/dnn/src/CMakeLists.txt index be9d48c5..bbfc8a2f 100644 --- a/dnn/src/CMakeLists.txt +++ b/dnn/src/CMakeLists.txt @@ -20,6 +20,7 @@ if(NOT ${MGE_ARCH} STREQUAL "naive") endif() endif() + if(MGE_WITH_CUDA) file(GLOB_RECURSE SOURCES_ cuda/*.cpp) list(APPEND SOURCES ${SOURCES_})