CMakeLists.txt

set_property(GLOBAL PROPERTY TABLE_DEPS string_helper)
set(graphDir graph)
get_property(TABLE_DEPS GLOBAL PROPERTY TABLE_DEPS)
set_source_files_properties(
  ${graphDir}/graph_edge.cc PROPERTIES COMPILE_FLAGS
                                       ${DISTRIBUTE_COMPILE_FLAGS})
cc_library(graph_edge SRCS ${graphDir}/graph_edge.cc)
set_source_files_properties(
  ${graphDir}/graph_weighted_sampler.cc PROPERTIES COMPILE_FLAGS
                                                   ${DISTRIBUTE_COMPILE_FLAGS})
cc_library(
  WeightedSampler
  SRCS ${graphDir}/graph_weighted_sampler.cc
  DEPS graph_edge)
set_source_files_properties(
  ${graphDir}/graph_node.cc PROPERTIES COMPILE_FLAGS
                                       ${DISTRIBUTE_COMPILE_FLAGS})
cc_library(
  graph_node
  SRCS ${graphDir}/graph_node.cc
  DEPS WeightedSampler)
set_source_files_properties(
  memory_dense_table.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
set_source_files_properties(
  barrier_table.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
set_source_files_properties(
  common_graph_table.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})

get_property(RPC_DEPS GLOBAL PROPERTY RPC_DEPS)

set(PADDLE_LIB_THIRD_PARTY_PATH "${PADDLE_LIB}/third_party/")
include_directories(
  ${PADDLE_LIB_THIRD_PARTY_PATH}libmct/src/extern_libmct/libmct/include)

set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fopenmp")

set(TABLE_SRC memory_dense_table.cc barrier_table.cc common_graph_table.cc)
#set(EXTERN_DEP rocksdb)

cc_library(
  common_table
  SRCS ${TABLE_SRC}
  DEPS ${TABLE_DEPS}
       ${RPC_DEPS}
       graph_edge
       graph_node
       device_context
       string_helper
       simple_threadpool
       xxhash
       generator)

set_source_files_properties(
  tensor_accessor.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})

cc_library(
  tensor_accessor
  SRCS tensor_accessor.cc
  DEPS ${TABLE_DEPS} eigen3 ps_framework_proto device_context)
cc_library(
  tensor_table
  SRCS
  DEPS eigen3
       ps_framework_proto
       executor
       scope
       device_context
       tensor
       ${TABLE_DEPS})
set_source_files_properties(table.cc PROPERTIES COMPILE_FLAGS
                                                ${DISTRIBUTE_COMPILE_FLAGS})

set_source_files_properties(
  sparse_sgd_rule.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
set_source_files_properties(
  ctr_double_accessor.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
set_source_files_properties(
  ctr_accessor.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
set_source_files_properties(
  sparse_accessor.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
set_source_files_properties(
  ctr_dymf_accessor.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
set_source_files_properties(
  memory_sparse_table.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
set_source_files_properties(
  ssd_sparse_table.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
set_source_files_properties(
  memory_sparse_geo_table.cc PROPERTIES COMPILE_FLAGS
                                        ${DISTRIBUTE_COMPILE_FLAGS})

cc_library(
  sparse_sgd_rule
  SRCS sparse_sgd_rule.cc
  DEPS ${TABLE_DEPS} ps_framework_proto)
cc_library(
  ctr_accessor
  SRCS ctr_accessor.cc ctr_double_accessor.cc sparse_accessor.cc
       ctr_dymf_accessor.cc
  DEPS ${TABLE_DEPS} ps_framework_proto sparse_sgd_rule)
cc_library(
  sparse_table
  SRCS memory_sparse_table.cc ssd_sparse_table.cc memory_sparse_geo_table.cc
  DEPS ps_framework_proto
       ${TABLE_DEPS}
       fs
       afs_wrapper
       ctr_accessor
       common_table
       rocksdb)

cc_library(
  table
  SRCS table.cc
  DEPS sparse_table
       common_table
       tensor_accessor
       tensor_table
       ps_framework_proto
       string_helper
       device_context
       gflags
       glog)

target_link_libraries(table -fopenmp)