From 3c6163868bfa05eb6cf11d1c9c851e344efa0b73 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Sun, 26 Apr 2020 16:44:01 +0800 Subject: [PATCH] fix(dnn): add cmake support for midout GitOrigin-RevId: 58fcbed88a846dfff029d6bd1369ea731171c806 --- CMakeLists.txt | 1 + dnn/src/CMakeLists.txt | 1 + 2 files changed, 2 insertions(+) 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_}) -- GitLab