CMakeLists.txt 1.6 KB
Newer Older
1
# compatible utils used for fluid op system
2 3
add_subdirectory(compat)

4
cc_library(errors SRCS errors.cc)
5
set(phi_enforce_deps errors flags)
X
xiongkun 已提交
6
if(WITH_GPU)
7
  set(phi_enforce_deps ${phi_enforce_deps} external_error_proto)
X
xiongkun 已提交
8
endif()
9
cc_library(phi_enforce INTERFACE SRCS enforce.cc DEPS ${phi_enforce_deps})
10

11 12
cc_library(kernel_factory SRCS kernel_factory.cc DEPS phi_enforce fluid_convert_utils)
cc_library(kernel_context SRCS kernel_context.cc DEPS phi_enforce phi_context)
13

14 15 16 17
cc_library(ddim SRCS ddim.cc DEPS phi_enforce)
cc_library(tensor_base SRCS tensor_base.cc allocator.cc DEPS phi_enforce)
cc_library(tensor_meta SRCS tensor_meta.cc DEPS phi_enforce)
cc_library(lod_utils SRCS lod_utils.cc DEPS phi_enforce)
18

19
cc_library(dense_tensor SRCS dense_tensor.cc dense_tensor_impl.cc DEPS fluid_convert_utils tensor_meta tensor_base)
20 21
cc_library(sparse_coo_tensor SRCS sparse_coo_tensor.cc DEPS tensor_meta tensor_base)
cc_library(sparse_csr_tensor SRCS sparse_csr_tensor.cc DEPS dense_tensor tensor_base)
J
Jack Zhou 已提交
22
cc_library(string_tensor SRCS string_tensor.cc DEPS convert_utils tensor_meta tensor_base)
23

24
cc_library(meta_tensor SRCS meta_tensor.cc DEPS tensor_base tensor_meta dense_tensor)
25
cc_library(infermeta_utils SRCS infermeta_utils.cc DEPS meta_tensor)
26
cc_library(selected_rows SRCS selected_rows_impl.cc DEPS dense_tensor phi_enforce ddim memcpy)
27
cc_library(phi_device_context SRCS device_context.cc DEPS dense_tensor selected_rows)
28

29
cc_library(custom_kernel SRCS custom_kernel.cc DEPS kernel_factory)
30

31 32 33
# Will remove once we implemented MKLDNN_Tensor
if(WITH_MKLDNN)
    add_dependencies(dense_tensor mkldnn)
W
Wilber 已提交
34
    add_dependencies(tensor_base mkldnn)
35
endif()