2022-09-30 10:02:52 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
set (EXTRA_LIBS "")
|
|
|
|
|
if(MNN_CUDA_PROFILE)
|
|
|
|
|
set(CUDA_MIN_VERSION "9.0")
|
|
|
|
|
find_package(CUDA ${CUDA_MIN_VERSION})
|
|
|
|
|
|
|
|
|
|
MESSAGE("cuda dir is:${CUDA_INCLUDE_DIRS}")
|
|
|
|
|
include_directories(
|
|
|
|
|
${CUDA_INCLUDE_DIRS}
|
|
|
|
|
)
|
|
|
|
|
set(EXTRA_LIBS ${CUDA_INCLUDE_DIRS}/../lib/libnvToolsExt.so)
|
|
|
|
|
endif()
|
|
|
|
|
|
2020-11-05 16:41:56 +08:00
|
|
|
file(GLOB_RECURSE MNN_EXPR_SRCS "${CMAKE_CURRENT_LIST_DIR}/*.*")
|
|
|
|
|
option(MNN_EXPR_SHAPE_EAGER "Force compute Expr's shape directly cost" OFF)
|
|
|
|
|
IF (MNN_EXPR_SHAPE_EAGER)
|
|
|
|
|
add_definitions(-DMNN_EXPR_SHAPE_EAGER)
|
|
|
|
|
ENDIF()
|
2020-01-17 10:20:15 +08:00
|
|
|
IF(MNN_SEP_BUILD)
|
2020-02-26 09:57:17 +08:00
|
|
|
if (MNN_BUILD_FOR_ANDROID_COMMAND)
|
|
|
|
|
set(CMAKE_LIBRARY_OUTPUT_DIRECTORY "../")
|
|
|
|
|
endif()
|
2025-05-23 15:21:41 +08:00
|
|
|
add_library(MNN_Express SHARED ${MNN_EXPR_SRCS})
|
2022-09-30 10:02:52 +08:00
|
|
|
target_link_libraries(MNN_Express MNN ${EXTRA_LIBS})
|
2022-02-18 11:30:27 +08:00
|
|
|
install(TARGETS MNN_Express
|
|
|
|
|
LIBRARY DESTINATION lib
|
|
|
|
|
ARCHIVE DESTINATION lib
|
|
|
|
|
)
|
2020-01-17 10:20:15 +08:00
|
|
|
ELSE()
|
2021-01-06 16:29:37 +08:00
|
|
|
add_library(MNN_Express OBJECT ${MNN_EXPR_SRCS})
|
2020-01-17 10:20:15 +08:00
|
|
|
ENDIF()
|
2022-09-30 10:02:52 +08:00
|
|
|
|
|
|
|
|
|
