Huawei_Technology
/
mindspore
Not watched
1
0
Fork 0
Code
Releases 13 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: 31f3611f9a
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '31f3611f9a'
${ noResults }
mindspore/predict/src/CMakeLists.txt

70 lines
2.6 kB
Raw Blame History 

  1. cmake_minimum_required(VERSION 3.12)

  2. project(mspredict)

  3. 

  4. set(CMAKE_CXX_STANDARD 11)

  5. 

  6. include_directories(${CMAKE_CURRENT_SOURCE_DIR}/..)

  7. include_directories(${CMAKE_CURRENT_SOURCE_DIR}/../module/tvm_kernel/incubator-tvm/3rdparty/dlpack/include/)

  8. 

  9. link_directories(${CMAKE_CURRENT_SOURCE_DIR}/../output/lib/)

  10. 

  11. if (ENABLE_ASAN)

  12.   set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -o0 -fsanitize=address -fno-omit-frame-pointer -fsanitize=undefined")

  13. endif()

  14. 

  15. set(MSPREDICT_SRC

  16.         runtime/allocator.cc

  17.         runtime/allocator.h

  18.         runtime/thread_pool.cc

  19.         runtime/thread_pool.h

  20.         runtime/workspace_pool.cc

  21.         runtime/workspace_pool.h

  22.         runtime/runtime_api.cc

  23.         runtime/runtime_api.h

  24.         context.cc

  25.         graph.cc

  26.         graph.h

  27.         graph_execution.cc

  28.         graph_execution.h

  29.         node.cc

  30.         node.h

  31.         op.cc

  32.         op.h

  33.         op_factory.cc

  34.         op_factory.h

  35.         op_registry.cc

  36.         op_registry.h

  37.         session.cc

  38.         tensor.cc

  39.         ${CMAKE_CURRENT_SOURCE_DIR}/operator/cpu/common/op_func_comm.cc)

  40. 

  41. set(MSPREDICT_SRC ${MSPREDICT_SRC}

  42. 	       ${CMAKE_CURRENT_SOURCE_DIR}/../common/graph_util.cc

  43. 	       ${CMAKE_CURRENT_SOURCE_DIR}/../common/utils.cc

  44. 	       ${CMAKE_CURRENT_SOURCE_DIR}/../common/mslog.cc

  45. 	       ${CMAKE_CURRENT_SOURCE_DIR}/../common/module_registry.cc)

  46. 

  47. add_library(mspredict SHARED ${MSPREDICT_SRC})

  48. 

  49. if(ENABLE_PREDICT_ARM64 OR ENABLE_PREDICT_ARM32)

  50.   target_link_libraries(mspredict android log tvm_kernel libsecurec.a)

  51. else()

  52.   target_link_libraries(mspredict pthread tvm_kernel libsecurec.a)

  53. endif()

  54. 

  55. add_dependencies(mspredict tvm_kernel)

  56. add_dependencies(mspredict securec)

  57. add_dependencies(mspredict gtest)

  58. 

  59. add_custom_command(TARGET mspredict POST_BUILD

  60.         COMMAND mkdir -pv ${PREDICT_DIR}/output/lib

  61.         COMMAND cp ${PREDICT_BUILD_DIR}/src/libmspredict.so ${PREDICT_DIR}/output/lib/

  62.         COMMAND cp ${PREDICT_BUILD_DIR}/module/tvm_kernel/lite/libtvm_kernel.so ${PREDICT_DIR}/output/lib/

  63.         COMMAND mkdir -pv ${PREDICT_DIR}/output/include

  64.         COMMAND cp -r ${PREDICT_DIR}/include/* ${PREDICT_DIR}/output/include

  65.         COMMAND mkdir -pv ${PREDICT_DIR}/output/include/schema/inner

  66.         COMMAND cp ${PREDICT_DIR}/schema/ms_generated.h ${PREDICT_DIR}/output/include/schema/inner

  67.         COMMAND cp ${PREDICT_DIR}/schema/op_generated.h ${PREDICT_DIR}/output/include/schema/inner

  68.         COMMAND mkdir -pv ${PREDICT_DIR}/output/include/dlpack/

  69.         COMMAND cp ${PREDICT_DIR}/module/tvm_kernel/incubator-tvm/3rdparty/dlpack/include/dlpack/dlpack.h ${PREDICT_DIR}/output/include/dlpack/)