CMakeLists.txt 871 字节
Newer Older
1 2 3
# paddle experimental common components
add_subdirectory(common)

4 5
# pten (low level) api headers: include
# pten (high level) api
6 7 8
add_subdirectory(api)
# pten core components
add_subdirectory(core)
C
Chen Weihang 已提交
9 10
# pten components of specific backends
add_subdirectory(backends)
11 12
# pten kernels for diff device
add_subdirectory(kernels)
C
Chen Weihang 已提交
13 14
# pten infermeta
add_subdirectory(infermeta)
C
Chen Weihang 已提交
15 16
# pten operator definitions
add_subdirectory(ops)
17 18
# pten tests
add_subdirectory(tests)
19 20

# make an unity target for compile deps
21
set(PTEN_DEPS convert_utils dense_tensor pten_context kernel_factory kernel_context arg_map_context infermeta lod_utils)
22
get_property(pten_kernels GLOBAL PROPERTY PTEN_KERNELS)
23 24
# keep this message for debug, remove it later if needless
message(STATUS "All standard pten kernels: ${pten_kernels}")
25
set(PTEN_DEPS ${PTEN_DEPS} ${pten_kernels})
26

27
cc_library(pten DEPS ${PTEN_DEPS})