CMakeLists.txt 3.8 KB
Newer Older
T
tangwei12 已提交
1
set_property(GLOBAL PROPERTY TABLE_DEPS string_helper)
S
seemingwang 已提交
2
set(graphDir graph)
T
tangwei12 已提交
3
get_property(TABLE_DEPS GLOBAL PROPERTY TABLE_DEPS)
4 5 6
set_source_files_properties(
  ${graphDir}/graph_edge.cc PROPERTIES COMPILE_FLAGS
                                       ${DISTRIBUTE_COMPILE_FLAGS})
S
seemingwang 已提交
7
cc_library(graph_edge SRCS ${graphDir}/graph_edge.cc)
8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27
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})
T
tangwei12 已提交
28

T
Thunderbrook 已提交
29 30
get_property(RPC_DEPS GLOBAL PROPERTY RPC_DEPS)

31
set(PADDLE_LIB_THIRD_PARTY_PATH "${PADDLE_LIB}/third_party/")
32 33
include_directories(
  ${PADDLE_LIB_THIRD_PARTY_PATH}libmct/src/extern_libmct/libmct/include)
34 35 36

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

Z
zhaocaibei123 已提交
37 38
set(TABLE_SRC memory_dense_table.cc barrier_table.cc common_graph_table.cc)
#set(EXTERN_DEP rocksdb)
T
Thunderbrook 已提交
39

40 41 42 43 44 45 46 47 48 49 50 51
cc_library(
  common_table
  SRCS ${TABLE_SRC}
  DEPS ${TABLE_DEPS}
       ${RPC_DEPS}
       graph_edge
       graph_node
       device_context
       string_helper
       simple_threadpool
       xxhash
       generator)
T
tangwei12 已提交
52

53 54
set_source_files_properties(
  tensor_accessor.cc PROPERTIES COMPILE_FLAGS ${DISTRIBUTE_COMPILE_FLAGS})
T
tangwei12 已提交
55

56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71
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})
T
tangwei12 已提交
72

73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89
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})
Z
zhaocaibei123 已提交
90

91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109
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)
110

111 112 113 114 115 116 117 118 119 120 121 122 123
cc_library(
  table
  SRCS table.cc
  DEPS sparse_table
       common_table
       tensor_accessor
       tensor_table
       ps_framework_proto
       string_helper
       device_context
       gflags
       glog
       boost)
124 125

target_link_libraries(table -fopenmp)