diff --git a/CMakeLists.txt b/CMakeLists.txt index 400a577a7ae890a6ebd2ac6eb28b2ea4dce84677..3b331d4aae6b0459d296b381348f96e441edf569 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 be9d48c510ee97622abe1c37a4aafae22017b585..bbfc8a2f233c955b98153a24b50287bc66273ad6 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_})