CMakeLists.txt 782 字节
Newer Older
J
Jack Zhou 已提交
1 2 3 4 5
add_subdirectory(cpu)
if(WITH_GPU OR WITH_ROCM)
  add_subdirectory(gpu)
endif()

6 7 8 9
cc_library(
  unicode
  SRCS unicode.cc
  DEPS utf8proc)
J
Jack Zhou 已提交
10 11
set_property(GLOBAL PROPERTY STRING_KERNELS "")

12 13 14 15 16 17 18 19 20 21 22
set(STRING_KERNEL_DEPS
    dense_tensor
    string_tensor
    sparse_coo_tensor
    sparse_csr_tensor
    kernel_context
    kernel_factory
    arg_map_context
    convert_utils
    lod_utils
    custom_kernel)
J
Jack Zhou 已提交
23 24 25 26 27 28 29
set(STRING_KERNEL_DEPS ${STRING_KERNEL_DEPS} eigen_function blas math_function)
# remove this dep after removing fluid deps on tensor creation
set(STRING_KERNEL_DEPS ${STRING_KERNEL_DEPS} phi_api_utils)
set(STRING_KERNEL_DEPS ${STRING_KERNEL_DEPS} string_infermeta)
set(STRING_KERNEL_DEPS ${STRING_KERNEL_DEPS} unicode)

register_kernels(DEPS ${STRING_KERNEL_DEPS} SUB_DIR "strings")