提交 3c616386 编写于 作者: M Megvii Engine Team 提交者: Xinran Xu

fix(dnn): add cmake support for midout

GitOrigin-RevId: 58fcbed88a846dfff029d6bd1369ea731171c806
上级 d345c862
......@@ -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)
......
......@@ -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_})
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册