Browse Source

!7773 [MS][LITE][Develop]support benchmark running on window10

Merge pull request !7773 from chenjianping/lite_windows
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
9ff6d86672
3 changed files with 16 additions and 8 deletions
  1. +7
    -7
      mindspore/lite/src/CMakeLists.txt
  2. +8
    -0
      mindspore/lite/src/ops/CMakeLists.txt
  3. +1
    -1
      mindspore/lite/tools/benchmark/CMakeLists.txt

+ 7
- 7
mindspore/lite/src/CMakeLists.txt View File

@@ -65,14 +65,14 @@ if (SUPPORT_TRAIN)
)
endif ()

file(GLOB_RECURSE C_OPS_SRC ${CMAKE_CURRENT_SOURCE_DIR}/ops/*.cc ${CMAKE_CURRENT_SOURCE_DIR}/ops/populate/*.cc)
add_subdirectory(ops)
add_subdirectory(runtime/kernel/arm)

add_library(mindspore-lite SHARED ${LITE_SRC} ${C_OPS_SRC})
add_library(mindspore-lite SHARED ${LITE_SRC})
add_dependencies(mindspore-lite fbs_src)
set_target_properties(mindspore-lite PROPERTIES CLEAN_DIRECT_OUTPUT 1)

add_library(mindspore-lite_static STATIC ${LITE_SRC} ${C_OPS_SRC})
add_library(mindspore-lite_static STATIC ${LITE_SRC})
add_dependencies(mindspore-lite_static fbs_src)
set_target_properties(mindspore-lite_static PROPERTIES OUTPUT_NAME "mindspore-lite")
set_target_properties(mindspore-lite_static PROPERTIES CLEAN_DIRECT_OUTPUT 1)
@@ -80,11 +80,11 @@ set_target_properties(mindspore-lite_static PROPERTIES CLEAN_DIRECT_OUTPUT 1)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wno-unused-private-field")
if (SUPPORT_GPU)
add_subdirectory(runtime/kernel/opencl)
target_link_libraries(mindspore-lite cpu_kernel_mid opencl_kernel_mid nnacl)
target_link_libraries(mindspore-lite_static cpu_kernel_mid opencl_kernel_mid nnacl_mid)
target_link_libraries(mindspore-lite cpu_kernel_mid opencl_kernel_mid nnacl cpu_ops_mid)
target_link_libraries(mindspore-lite_static cpu_kernel_mid opencl_kernel_mid nnacl_mid cpu_ops_mid)
else ()
target_link_libraries(mindspore-lite cpu_kernel_mid nnacl)
target_link_libraries(mindspore-lite_static cpu_kernel_mid nnacl_mid)
target_link_libraries(mindspore-lite cpu_kernel_mid nnacl cpu_ops_mid)
target_link_libraries(mindspore-lite_static cpu_kernel_mid nnacl_mid cpu_ops_mid)
endif ()
if (PLATFORM_ARM32 OR PLATFORM_ARM64)
target_link_libraries(mindspore-lite log)


+ 8
- 0
mindspore/lite/src/ops/CMakeLists.txt View File

@@ -0,0 +1,8 @@
include_directories(${CMAKE_CURRENT_SOURCE_DIR}/)

file(GLOB OPS_SRC
${CMAKE_CURRENT_SOURCE_DIR}/*.cc
${CMAKE_CURRENT_SOURCE_DIR}/populate/*.cc
)

add_library(cpu_ops_mid OBJECT ${OPS_SRC})

+ 1
- 1
mindspore/lite/tools/benchmark/CMakeLists.txt View File

@@ -19,7 +19,7 @@ if (PLATFORM_ARM32 OR PLATFORM_ARM64)
target_link_libraries(benchmark mindspore-lite)
else()
if (WIN32)
target_link_libraries(benchmark mindspore-lite_static pthread cpu_kernel_mid nnacl_mid)
target_link_libraries(benchmark mindspore-lite_static pthread cpu_kernel_mid nnacl_mid cpu_ops_mid)
else ()
target_link_libraries(benchmark mindspore-lite pthread)
endif ()


Loading…
Cancel
Save