CMakeLists.txt 2.1 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11
if(NOT WITH_DISTRIBUTE)
    return()
endif()

if(WITH_GRPC)
    set(cc_generic_services "false")
else()
    set(cc_generic_services "true")
endif()
configure_file(send_recv.proto.in ${CMAKE_CURRENT_SOURCE_DIR}/send_recv.proto @ONLY)

G
gongweibao 已提交
12
if(WITH_GRPC)
13 14 15 16
  grpc_library(sendrecvop_grpc SRCS grpc_bytebuffer_stream.cc sendrecvop_utils.cc grpc_client.cc
        request_handler_impl.cc rpc_client.cc rpc_server.cc grpc_server.cc variable_response.cc grpc_variable_response.cc grpc_serde.cc
      PROTO send_recv.proto 
      DEPS lod_tensor selected_rows memory)
17
  set(DISTRIBUTE_COMPILE_FLAGS "-Wno-non-virtual-dtor -Wno-error=non-virtual-dtor -Wno-error=delete-non-virtual-dtor")
G
gongweibao 已提交
18
  set_source_files_properties(grpc_serde_test.cc rpc_server_test.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
19
  cc_test(grpc_serde_test SRCS grpc_serde_test.cc 
20 21 22
    DEPS grpc++_unsecure grpc_unsecure gpr cares zlib protobuf sendrecvop_grpc scope profiler math_function SERIAL)
  cc_test(rpc_server_test SRCS rpc_server_test.cc
    DEPS sendrecvop_grpc grpc++_unsecure grpc_unsecure gpr cares zlib protobuf executor  proto_desc lookup_sparse_table_op SERIAL)
G
gongweibao 已提交
23
  cc_test(varhandle_test SRCS varhandle_test.cc DEPS profiler)
G
gongweibao 已提交
24
  return()
25
endif()
G
gongweibao 已提交
26 27 28


set(DISTRIBUTE_COMPILE_FLAGS "-Wno-non-virtual-dtor -Wno-error=non-virtual-dtor -Wno-error=delete-non-virtual-dtor")
29 30 31 32 33 34

set_source_files_properties(brpc_server.cc brpc_client.cc rpc_server_test.cc brpc_serde_test.cc
    brpc_variable_response.cc brpc_sendrecvop_utils.cc brpc_rdma_pool.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})

brpc_library(sendrecvop_brpc SRCS brpc_client.cc brpc_server.cc rpc_server.cc rpc_client.cc request_handler_impl.cc brpc_sendrecvop_utils.cc 
    brpc_variable_response.cc variable_response.cc sendrecvop_utils.cc brpc_rdma_pool.cc
G
gongweibao 已提交
35 36 37
  PROTO send_recv.proto
  DEPS lod_tensor selected_rows memory)

38
set(brpc_test_depends sendrecvop_brpc brpc ssl crypto protobuf leveldb gflags glog executor proto_desc lookup_table_op snappystream snappy)
G
gongweibao 已提交
39

40 41
cc_test(brpc_server_test SRCS rpc_server_test.cc 
    DEPS ${brpc_test_depends} SERIAL)
G
gongweibao 已提交
42

43 44
cc_test(brpc_serde_test SRCS brpc_serde_test.cc 
    DEPS ${brpc_test_depends} SERIAL)