98 lines
3.9 KiB
CMake
98 lines
3.9 KiB
CMake
set(MNN_CPP_TOOLS "")
|
|
if(CMAKE_SYSTEM_PROCESSOR MATCHES "^armv7" OR ARCHS MATCHES "^armv7(;armv7s)?")
|
|
add_definitions(-DMNN_USE_NEON)
|
|
elseif(CMAKE_SYSTEM_PROCESSOR MATCHES "^aarch64" OR ARCHS STREQUAL "arm64")
|
|
add_definitions(-DMNN_USE_NEON)
|
|
endif()
|
|
add_executable(MNNV2Basic.out ${CMAKE_CURRENT_LIST_DIR}/MNNV2Basic.cpp)
|
|
list(APPEND MNN_CPP_TOOLS MNNV2Basic.out)
|
|
|
|
if (NOT MNN_SKIPBUILD_GEOMETRY)
|
|
add_executable(GetMNNInfo ${CMAKE_CURRENT_LIST_DIR}/GetMNNInfo.cpp)
|
|
list(APPEND MNN_CPP_TOOLS GetMNNInfo)
|
|
add_executable(ModuleBasic.out ${CMAKE_CURRENT_LIST_DIR}/ModuleBasic.cpp)
|
|
list(APPEND MNN_CPP_TOOLS ModuleBasic.out)
|
|
|
|
IF(CMAKE_SYSTEM_NAME MATCHES "^Android")
|
|
add_executable(GpuInterTest.out ${CMAKE_CURRENT_LIST_DIR}/GpuInterTest.cpp )
|
|
target_link_libraries(GpuInterTest.out android)
|
|
list(APPEND MNN_CPP_TOOLS GpuInterTest.out)
|
|
ENDIF()
|
|
|
|
add_executable(OpenCLProgramBuildTest.out ${CMAKE_CURRENT_LIST_DIR}/OpenCLProgramBuildTest.cpp )
|
|
list(APPEND MNN_CPP_TOOLS OpenCLProgramBuildTest.out)
|
|
|
|
add_executable(SequenceModuleTest.out ${CMAKE_CURRENT_LIST_DIR}/SequenceModuleTest.cpp)
|
|
list(APPEND MNN_CPP_TOOLS SequenceModuleTest.out)
|
|
|
|
add_executable(mergeInplaceForCPU ${CMAKE_CURRENT_LIST_DIR}/mergeInplaceForCPU.cpp)
|
|
list(APPEND MNN_CPP_TOOLS mergeInplaceForCPU)
|
|
|
|
add_executable(modelCompare.out ${CMAKE_CURRENT_LIST_DIR}/modelCompare.cpp)
|
|
list(APPEND MNN_CPP_TOOLS modelCompare.out)
|
|
|
|
if (MNN_USE_SSE)
|
|
target_compile_options(MNNV2Basic.out PRIVATE -DMNN_USE_SSE)
|
|
endif()
|
|
|
|
add_executable(mobilenetTest.out ${CMAKE_CURRENT_LIST_DIR}/mobilenetTest.cpp )
|
|
list(APPEND MNN_CPP_TOOLS mobilenetTest.out)
|
|
|
|
add_executable(backendTest.out ${CMAKE_CURRENT_LIST_DIR}/backendTest.cpp)
|
|
list(APPEND MNN_CPP_TOOLS backendTest.out)
|
|
|
|
add_executable(testModel.out ${CMAKE_CURRENT_LIST_DIR}/testModel.cpp)
|
|
list(APPEND MNN_CPP_TOOLS testModel.out)
|
|
|
|
add_executable(compilefornpu ${CMAKE_CURRENT_LIST_DIR}/compilefornpu.cpp ${CMAKE_CURRENT_LIST_DIR}/../../3rd_party/flatbuffers/src/util.cpp)
|
|
list(APPEND MNN_CPP_TOOLS compilefornpu)
|
|
|
|
add_executable(testModel_expr.out ${CMAKE_CURRENT_LIST_DIR}/testModel_expr.cpp)
|
|
list(APPEND MNN_CPP_TOOLS testModel_expr.out)
|
|
|
|
add_executable(testModelWithDescribe.out ${CMAKE_CURRENT_LIST_DIR}/testModelWithDescribe.cpp)
|
|
list(APPEND MNN_CPP_TOOLS testModelWithDescribe.out)
|
|
|
|
add_executable(getPerformance.out ${CMAKE_CURRENT_LIST_DIR}/getPerformance.cpp)
|
|
list(APPEND MNN_CPP_TOOLS getPerformance.out)
|
|
|
|
add_executable(checkInvalidValue.out ${CMAKE_CURRENT_LIST_DIR}/checkInvalidValue.cpp)
|
|
list(APPEND MNN_CPP_TOOLS checkInvalidValue.out)
|
|
|
|
add_executable(timeProfile.out ${CMAKE_CURRENT_LIST_DIR}/timeProfile.cpp ${CMAKE_CURRENT_LIST_DIR}/revertMNNModel.cpp ${CMAKE_CURRENT_LIST_DIR}/Profiler.cpp)
|
|
list(APPEND MNN_CPP_TOOLS timeProfile.out)
|
|
|
|
add_executable(testTrain.out ${CMAKE_CURRENT_LIST_DIR}/testTrain.cpp)
|
|
list(APPEND MNN_CPP_TOOLS testTrain.out)
|
|
|
|
add_executable(fuseTest ${CMAKE_CURRENT_LIST_DIR}/fuseTest.cpp)
|
|
list(APPEND MNN_CPP_TOOLS fuseTest)
|
|
|
|
if (MNN_QNN)
|
|
add_executable(generateIO ${CMAKE_CURRENT_LIST_DIR}/generateIO.cpp)
|
|
list(APPEND MNN_CPP_TOOLS generateIO)
|
|
add_executable(MNN2QNNModel ${CMAKE_CURRENT_LIST_DIR}/MNN2QNNModel.cpp)
|
|
list(APPEND MNN_CPP_TOOLS MNN2QNNModel)
|
|
endif()
|
|
|
|
endif()
|
|
|
|
foreach(TARGET ${MNN_CPP_TOOLS})
|
|
target_link_libraries(${TARGET} ${MNN_DEPS})
|
|
if (MSVC)
|
|
target_compile_definitions(${TARGET} PRIVATE "_CRT_SECURE_NO_WARNINGS")
|
|
if (NOT MNN_BUILD_SHARED_LIBS)
|
|
foreach (DEPEND ${MNN_DEPS})
|
|
target_link_options(${TARGET} PRIVATE /WHOLEARCHIVE:$<TARGET_FILE:${DEPEND}>)
|
|
endforeach ()
|
|
endif()
|
|
endif()
|
|
endforeach()
|
|
|
|
if (NOT WIN32)
|
|
add_executable(checkDir.out ${CMAKE_CURRENT_LIST_DIR}/checkDir.cpp)
|
|
add_executable(checkFile.out ${CMAKE_CURRENT_LIST_DIR}/checkFile.cpp)
|
|
add_executable(winogradExample.out ${CMAKE_CURRENT_LIST_DIR}/winogradExample.cpp)
|
|
target_link_libraries(winogradExample.out ${MNN_DEPS})
|
|
endif()
|