Browse Source

fix(dnn): add cmake support for midout

GitOrigin-RevId: 58fcbed88a
tags/v0.4.0
Megvii Engine Team Xinran Xu 5 years ago
parent
commit
3c6163868b
2 changed files with 2 additions and 0 deletions
  1. +1
    -0
      CMakeLists.txt
  2. +1
    -0
      dnn/src/CMakeLists.txt

+ 1
- 0
CMakeLists.txt View File

@@ -397,6 +397,7 @@ endif()


set (CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${MARCH}") set (CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${MARCH}")



# Write out megbrain_build_config.h # Write out megbrain_build_config.h
# It defines macros needed by both megbrain and dnn # 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) configure_file(src/megbrain_build_config.h.in ${CMAKE_CURRENT_BINARY_DIR}/genfiles/megbrain_build_config.h)


+ 1
- 0
dnn/src/CMakeLists.txt View File

@@ -20,6 +20,7 @@ if(NOT ${MGE_ARCH} STREQUAL "naive")
endif() endif()
endif() endif()



if(MGE_WITH_CUDA) if(MGE_WITH_CUDA)
file(GLOB_RECURSE SOURCES_ cuda/*.cpp) file(GLOB_RECURSE SOURCES_ cuda/*.cpp)
list(APPEND SOURCES ${SOURCES_}) list(APPEND SOURCES ${SOURCES_})


Loading…
Cancel
Save