• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1set(grpc_USE_STATIC_LIBS OFF)
2if(${CMAKE_SYSTEM_NAME} MATCHES "Darwin")
3    set(grpc_CXXFLAGS "-fstack-protector-all -Wno-uninitialized -Wno-unused-parameter -fPIC -D_FORTIFY_SOURCE=2 -O2 \
4        -Dgrpc=mindspore_grpc -Dgrpc_impl=mindspore_grpc_impl -Dgrpc_core=mindspore_grpc_core")
5elseif(${CMAKE_SYSTEM_NAME} MATCHES "Windows")
6    set(grpc_CXXFLAGS "-fstack-protector-all -Wno-maybe-uninitialized -Wno-unused-parameter -D_FORTIFY_SOURCE=2 -O2")
7else()
8    set(grpc_CXXFLAGS "-fstack-protector-all -Wno-maybe-uninitialized -Wno-unused-parameter -D_FORTIFY_SOURCE=2 -O2 \
9        -Dgrpc=mindspore_grpc -Dgrpc_impl=mindspore_grpc_impl -Dgrpc_core=mindspore_grpc_core")
10    set(grpc_CFLAGS "-fstack-protector-all -D_FORTIFY_SOURCE=2 -O2")
11    if(NOT ENABLE_GLIBCXX)
12        set(grpc_CXXFLAGS "${grpc_CXXFLAGS} -D_GLIBCXX_USE_CXX11_ABI=0")
13    endif()
14endif()
15
16if(NOT ${CMAKE_SYSTEM_NAME} MATCHES "Darwin")
17    set(grpc_LDFLAGS "-Wl,-z,relro,-z,now,-z,noexecstack")
18endif()
19
20if(EXISTS ${protobuf_ROOT}/lib64)
21    set(_FINDPACKAGE_PROTOBUF_CONFIG_DIR "${protobuf_ROOT}/lib64/cmake/protobuf")
22else()
23    set(_FINDPACKAGE_PROTOBUF_CONFIG_DIR "${protobuf_ROOT}/lib/cmake/protobuf")
24endif()
25message("grpc using Protobuf_DIR : " ${_FINDPACKAGE_PROTOBUF_CONFIG_DIR})
26
27if(EXISTS ${absl_ROOT}/lib64)
28    set(_FINDPACKAGE_ABSL_CONFIG_DIR "${absl_ROOT}/lib64/cmake/absl")
29else()
30    set(_FINDPACKAGE_ABSL_CONFIG_DIR "${absl_ROOT}/lib/cmake/absl")
31endif()
32message("grpc using absl_DIR : " ${_FINDPACKAGE_ABSL_CONFIG_DIR})
33
34if(EXISTS ${c-ares_ROOT}/lib64)
35    set(_FINDPACKAGE_CARES_CONFIG_DIR "${c-ares_ROOT}/lib64/cmake/c-ares")
36else()
37    set(_FINDPACKAGE_CARES_CONFIG_DIR "${c-ares_ROOT}/lib/cmake/c-ares")
38endif()
39message("grpc using c-ares_DIR : " ${_FINDPACKAGE_CARES_CONFIG_DIR})
40
41if(EXISTS ${re2_ROOT}/lib64)
42    set(_FINDPACKAGE_RE2_CONFIG_DIR "${re2_ROOT}/lib64/cmake/re2")
43else()
44    set(_FINDPACKAGE_RE2_CONFIG_DIR "${re2_ROOT}/lib/cmake/re2")
45endif()
46message("grpc using re2_DIR : " ${_FINDPACKAGE_RE2_CONFIG_DIR})
47
48if(EXISTS ${openssl_ROOT})
49    set(_CMAKE_ARGS_OPENSSL_ROOT_DIR "-DOPENSSL_ROOT_DIR:PATH=${openssl_ROOT}")
50endif()
51
52if(ENABLE_GITEE)
53    set(REQ_URL "https://gitee.com/mirrors/grpc/repository/archive/v1.36.1.tar.gz")
54    set(SHA256 "17a3ac19345a6aeda01b2baba5400e1136b02b44770dbdfe8581255a091aaf87")
55else()
56    set(REQ_URL "https://github.com/grpc/grpc/archive/v1.36.1.tar.gz")
57    set(SHA256 "adf51558bf3d057a65651880c9814e09e77b61573eb950c2be1142a624d58e69")
58endif()
59
60mindspore_add_pkg(grpc
61        VER 1.36.1
62        LIBS mindspore_grpc++ mindspore_grpc mindspore_gpr mindspore_upb mindspore_address_sorting
63        EXE grpc_cpp_plugin
64        URL ${REQ_URL}
65        SHA256 ${SHA256}
66        PATCHES ${TOP_DIR}/third_party/patch/grpc/grpc.patch001
67        CMAKE_OPTION -DCMAKE_BUILD_TYPE:STRING=Release
68        -DBUILD_SHARED_LIBS=ON
69        -DgRPC_INSTALL:BOOL=ON
70        -DgRPC_BUILD_TESTS:BOOL=OFF
71        -DgRPC_PROTOBUF_PROVIDER:STRING=package
72        -DgRPC_PROTOBUF_PACKAGE_TYPE:STRING=CONFIG
73        -DProtobuf_DIR:PATH=${_FINDPACKAGE_PROTOBUF_CONFIG_DIR}
74        -DgRPC_ZLIB_PROVIDER:STRING=package
75        -DZLIB_ROOT:PATH=${zlib_ROOT}
76        -DgRPC_ABSL_PROVIDER:STRING=package
77        -Dabsl_DIR:PATH=${_FINDPACKAGE_ABSL_CONFIG_DIR}
78        -DgRPC_CARES_PROVIDER:STRING=package
79        -Dc-ares_DIR:PATH=${_FINDPACKAGE_CARES_CONFIG_DIR}
80        -DgRPC_SSL_PROVIDER:STRING=package
81        ${_CMAKE_ARGS_OPENSSL_ROOT_DIR}
82        -DgRPC_RE2_PROVIDER:STRING=package
83        -Dre2_DIR:PATH=${_FINDPACKAGE_RE2_CONFIG_DIR}
84        )
85
86include_directories(${grpc_INC})
87
88add_library(mindspore::grpc++ ALIAS grpc::mindspore_grpc++)
89
90# link other grpc libs
91target_link_libraries(grpc::mindspore_grpc++ INTERFACE grpc::mindspore_grpc grpc::mindspore_gpr grpc::mindspore_upb
92    grpc::mindspore_address_sorting)
93
94# modify mindspore macro define
95add_compile_definitions(grpc=mindspore_grpc)
96add_compile_definitions(grpc_impl=mindspore_grpc_impl)
97add_compile_definitions(grpc_core=mindspore_grpc_core)
98
99function(ms_grpc_generate c_var h_var)
100    if(NOT ARGN)
101        message(SEND_ERROR "Error: ms_grpc_generate() called without any proto files")
102        return()
103    endif()
104
105    set(${c_var})
106    set(${h_var})
107
108    foreach(file ${ARGN})
109        get_filename_component(abs_file ${file} ABSOLUTE)
110        get_filename_component(file_name ${file} NAME_WE)
111        get_filename_component(file_dir ${abs_file} PATH)
112        file(RELATIVE_PATH rel_path ${CMAKE_CURRENT_SOURCE_DIR} ${file_dir})
113
114        list(APPEND ${c_var} "${CMAKE_BINARY_DIR}/proto/${file_name}.pb.cc")
115        list(APPEND ${h_var} "${CMAKE_BINARY_DIR}/proto/${file_name}.pb.h")
116        list(APPEND ${c_var} "${CMAKE_BINARY_DIR}/proto/${file_name}.grpc.pb.cc")
117        list(APPEND ${h_var} "${CMAKE_BINARY_DIR}/proto/${file_name}.grpc.pb.h")
118
119        add_custom_command(
120                OUTPUT "${CMAKE_BINARY_DIR}/proto/${file_name}.pb.cc"
121                "${CMAKE_BINARY_DIR}/proto/${file_name}.pb.h"
122                "${CMAKE_BINARY_DIR}/proto/${file_name}.grpc.pb.cc"
123                "${CMAKE_BINARY_DIR}/proto/${file_name}.grpc.pb.h"
124                WORKING_DIRECTORY ${PROJECT_SOURCE_DIR}
125                COMMAND ${CMAKE_COMMAND} -E make_directory "${CMAKE_BINARY_DIR}/proto"
126                COMMAND protobuf::protoc --version
127                COMMAND protobuf::protoc -I${file_dir} --cpp_out=${CMAKE_BINARY_DIR}/proto
128                        --grpc_out=${CMAKE_BINARY_DIR}/proto
129                        --plugin=protoc-gen-grpc=$<TARGET_FILE:grpc::grpc_cpp_plugin> ${abs_file}
130                DEPENDS protobuf::protoc grpc::grpc_cpp_plugin ${abs_file}
131                COMMENT "Running C++ gRPC compiler on ${file}" VERBATIM)
132    endforeach()
133
134    set_source_files_properties(${${c_var}} ${${h_var}} PROPERTIES GENERATED TRUE)
135    set(${c_var} ${${c_var}} PARENT_SCOPE)
136    set(${h_var} ${${h_var}} PARENT_SCOPE)
137endfunction()
138