Huawei_Technology
/
MindSpore-Serving
Not watched
1
0
Fork 0
Code
Releases 3 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: ca0bb64cd0
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'ca0bb64cd0'
${ noResults }
MindSpore-Serving/tests/ut/CMakeLists.txt

102 lines
4.7 kB
Raw Blame History 

  1. 

  2. add_library(protobuf::libprotobuf ALIAS protobuf::protobuf)

  3. add_executable(protobuf::libprotoc ALIAS protobuf::protoc)

  4. 

  5. set(_PROTOBUF_LIBPROTOBUF protobuf::libprotobuf)

  6. set(_PROTOBUF_PROTOC $<TARGET_FILE:protobuf::protoc>)

  7. 

  8. # Find gRPC installation

  9. # Looks for gRPCConfig.cmake file installed by gRPC's cmake installation.

  10. if(EXISTS ${grpc_ROOT}/lib64)

  11.     set(gRPC_DIR "${grpc_ROOT}/lib64/cmake/grpc")

  12. else()

  13.     set(gRPC_DIR "${grpc_ROOT}/lib/cmake/grpc")

  14. endif()

  15. message("serving using grpc_DIR : " ${gPRC_DIR})

  16. 

  17. find_package(gRPC CONFIG REQUIRED)

  18. message(STATUS "Using gRPC ${gRPC_VERSION}")

  19. 

  20. set(_GRPC_GRPCPP gRPC::grpc++)

  21. set(_REFLECTION gRPC::grpc++_reflection)

  22. 

  23. set(_GRPC_CPP_PLUGIN_EXECUTABLE $<TARGET_FILE:gRPC::grpc_cpp_plugin>)

  24. set(_GRPC_PYTHON_PLUGIN_EXECUTABLE $<TARGET_FILE:gRPC::grpc_python_plugin>)

  25. 

  26. # Proto file

  27. # Generated sources

  28. file(MAKE_DIRECTORY "${CMAKE_CURRENT_BINARY_DIR}/proto/")

  29. file(GLOB_RECURSE PROTO_FILE_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} ../../mindspore_serving/proto/*.proto)

  30. foreach(proto_file_with_path ${PROTO_FILE_LIST})

  31.     message(------proto file: ${proto_file_with_path})

  32.     get_filename_component(proto_I_DIR "../../" ABSOLUTE)

  33.     get_filename_component(proto_file_absolute "${proto_file_with_path}" ABSOLUTE)

  34.     string(REGEX REPLACE .*/proto/ "" proto_file ${proto_file_with_path})

  35.     set(proto_file_relative "mindspore_serving/proto/${proto_file}")

  36.     string(REGEX REPLACE .proto "" proto_file_prefix ${proto_file})

  37. 

  38.     set(protoc_output_prefix ${CMAKE_CURRENT_BINARY_DIR}/mindspore_serving/proto)

  39.     set(hw_proto_srcs "${protoc_output_prefix}/${proto_file_prefix}.pb.cc")

  40.     set(hw_proto_hdrs "${protoc_output_prefix}/${proto_file_prefix}.pb.h")

  41.     set(hw_grpc_srcs "${protoc_output_prefix}/${proto_file_prefix}.grpc.pb.cc")

  42.     set(hw_grpc_hdrs "${protoc_output_prefix}/${proto_file_prefix}.grpc.pb.h")

  43.     set(hw_py_pb2 "${protoc_output_prefix}/${proto_file_prefix}_pb2.py")

  44.     set(hw_py_pb2_grpc "${protoc_output_prefix}/${proto_file_prefix}_pb2_grpc.py")

  45.     add_custom_command(

  46.             OUTPUT ${hw_proto_srcs} ${hw_proto_hdrs} ${hw_grpc_srcs} ${hw_grpc_hdrs} ${hw_py_pb2} ${hw_py_pb2_grpc}

  47.             WORKING_DIRECTORY ${proto_I_DIR}

  48.             COMMAND ${_PROTOBUF_PROTOC}

  49.             ARGS --grpc_out "${CMAKE_CURRENT_BINARY_DIR}"

  50.             --cpp_out "${CMAKE_CURRENT_BINARY_DIR}"

  51.             -I "${proto_I_DIR}"

  52.             --plugin=protoc-gen-grpc="${_GRPC_CPP_PLUGIN_EXECUTABLE}"

  53.             "${proto_file_relative}"

  54.             COMMAND ${_PROTOBUF_PROTOC}

  55.             ARGS --grpc_out "${CMAKE_CURRENT_BINARY_DIR}"

  56.             --python_out "${CMAKE_CURRENT_BINARY_DIR}"

  57.             -I "${proto_I_DIR}"

  58.             --plugin=protoc-gen-grpc="${_GRPC_PYTHON_PLUGIN_EXECUTABLE}"

  59.             "${proto_file_relative}"

  60.             DEPENDS "${proto_file_absolute}")

  61. 

  62.     list(APPEND PROTO_SRC_LIST ${hw_proto_srcs} ${hw_grpc_srcs})

  63. endforeach()

  64. 

  65. include_directories("${CMAKE_CURRENT_BINARY_DIR}/mindspore_serving") # for proto header file

  66. include_directories("${CMAKE_CURRENT_BINARY_DIR}")

  67. include_directories("${CMAKE_SOURCE_DIR}/mindspore_serving")

  68. 

  69. # serving_common for c++ server and python interface

  70. file(GLOB_RECURSE UT_SERVING_CORE_SRC RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}

  71.         "../../mindspore_serving/ccsrc/common/*.cc"

  72.         "../../mindspore_serving/ccsrc/master/*.cc"

  73.         "../../mindspore_serving/ccsrc/worker/*.cc")

  74. 

  75. file(GLOB_RECURSE UT_SERVING_STUB RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "stub/*.cc")

  76. set(UT_SERVING_COMMON ${UT_SERVING_CORE_SRC} ${UT_SERVING_STUB} ${PROTO_SRC_LIST})

  77. 

  78. include_directories(${CMAKE_CURRENT_SOURCE_DIR})

  79. include_directories(stub)

  80. include_directories(stub/include)

  81. include_directories(${CMAKE_SOURCE_DIR})

  82. include_directories(${CMAKE_CURRENT_SOURCE_DIR}/../../mindspore_serving/ccsrc)

  83. 

  84. link_directories(${CMKAE_BINARY_DIR}/securec/src)

  85. 

  86. add_library(serving_ut_common STATIC ${UT_SERVING_COMMON})

  87. 

  88. target_link_libraries(serving_ut_common PRIVATE ${_REFLECTION} ${_GRPC_GRPCPP} ${_PROTOBUF_LIBPROTOBUF})

  89. target_link_libraries(serving_ut_common PRIVATE mindspore_serving::event mindspore_serving::event_pthreads)

  90. target_link_libraries(serving_ut_common PRIVATE pthread mindspore_serving::glog)

  91. target_link_libraries(serving_ut_common PRIVATE ${SECUREC_LIBRARY})

  92. 

  93. # copy libevent lib

  94. file(GLOB_RECURSE LIBEVENT_LIB_LIST ${libevent_LIBPATH}/libevent* libevent_LIBPATH}/libevent_pthreads*)

  95. file(COPY ${LIBEVENT_LIB_LIST} DESTINATION ${CMAKE_CURRENT_BINARY_DIR}/lib)

  96. # copy glog lib

  97. file(GLOB_RECURSE GLOG_LIB_LIST ${glog_LIBPATH}/libglog*)

  98. file(COPY ${GLOG_LIB_LIST} DESTINATION ${CMAKE_CURRENT_BINARY_DIR}/lib)

  99. 

  100. add_subdirectory("cpp")

  101. add_subdirectory("python")