This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
Repositories
Datasets
Forum
实训
竞赛
大数据
应用
Register
Sign In
OpenI
/
MegEngine
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
31
Wiki
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
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
d345c86277
commit
3c6163868b
2 changed files
with
2 additions
and
0 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+1
-0
CMakeLists.txt
+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}")
# 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)
+ 1
- 0
dnn/src/CMakeLists.txt
View File
@@ -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_})
Write
Preview
Loading…
Cancel
Save