CMakeLists.txt 4.2 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11
set(kernel_declare_file
    ${PADDLE_BINARY_DIR}/paddle/phi/kernels/declarations.h.tmp
    CACHE INTERNAL "declarations.h file")
set(kernel_declare_file_final
    ${PADDLE_BINARY_DIR}/paddle/phi/kernels/declarations.h)
file(
  WRITE ${kernel_declare_file}
  "// Generated by the paddle/phi/kernels/CMakeLists.txt.  DO NOT EDIT!\n\n#pragma once\n\n"
)
file(APPEND ${kernel_declare_file}
     "#include \"paddle/phi/core/kernel_registry.h\"\n\n")
12

13
# phi functors and functions called by kernels
C
Chen Weihang 已提交
14
add_subdirectory(funcs)
C
Chen Weihang 已提交
15

16 17 18
# kernel autotune
add_subdirectory(autotune)

19 20
# phi depends all phi kernel targets
set_property(GLOBAL PROPERTY PHI_KERNELS "")
21

22
# [ 1. Common kernel compilation dependencies ]
23 24
set(COMMON_KERNEL_DEPS
    dense_tensor
25
    string_tensor
26 27
    sparse_coo_tensor
    sparse_csr_tensor
28
    tensor_array
29 30 31 32 33
    kernel_context
    kernel_factory
    arg_map_context
    convert_utils
    lod_utils
34
    custom_kernel
35
    string_infermeta
36
    phi_tensor_utils)
37 38 39 40 41 42 43 44 45
set(COMMON_KERNEL_DEPS
    ${COMMON_KERNEL_DEPS}
    eigen_function
    blas
    math_function
    im2col
    vol2col
    concat_and_split_functor
    selected_rows_functor)
46
# remove this dep after removing fluid deps on tensor creation
47
set(COMMON_KERNEL_DEPS ${COMMON_KERNEL_DEPS} phi_api_utils)
48 49
set(COMMON_KERNEL_DEPS ${COMMON_KERNEL_DEPS} infermeta infermeta_utils)
set(COMMON_KERNEL_DEPS ${COMMON_KERNEL_DEPS} switch_autotune)
50

51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67
set(COMMON_KERNEL_DEPS
    ${COMMON_KERNEL_DEPS}
    threadpool
    jit_kernel_helper
    softmax
    cross_entropy
    matrix_bit_code
    lapack_function
    lstm_compute
    gru_compute
    deformable_conv_functor
    matrix_reduce
    segment_pooling
    gather_scatter_kernel
    pooling
    maxouting
    matrix_inverse
68
    matrix_solve
69 70
    phi_dynload_warpctc
    sequence_padding
F
Feiyu Chan 已提交
71
    sequence_scale
72
    fft
73
    phi_data_layout_transform
74
    gpc
75 76
    utf8proc
    device_memory_aligment)
77

L
LiYuRio 已提交
78 79 80 81 82
set(COMMON_KERNEL_DEPS ${COMMON_KERNEL_DEPS} processgroup)
if(WITH_NCCL OR WITH_RCCL)
  set(COMMON_KERNEL_DEPS ${COMMON_KERNEL_DEPS} processgroup_nccl)
endif()

83
copy_if_different(${kernel_declare_file} ${kernel_declare_file_final})
84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100

file(GLOB kernel_h "*.h" "selected_rows/*.h" "sparse/*.h" "strings/*.h")
file(GLOB kernel_impl_h "impl/*.h" "selected_rows/impl/*.h")
file(GLOB kernel_primitive_h "primitive/*.h")

file(
  GLOB
  kernel_cu
  "gpu/*.cu"
  "gpu/*.cu.cc"
  "gpudnn/*.cu"
  "kps/*.cu"
  "selected_rows/gpu/*.cu"
  "sparse/gpu/*.cu"
  "strings/*.cu"
  "strings/gpu/*.cu")

101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127
if(WITH_MKLDNN)
  file(
    GLOB
    kernel_cc
    "*.cc"
    "cpu/*.cc"
    "selected_rows/*.cc"
    "selected_rows/cpu/*.cc"
    "sparse/*.cc"
    "sparse/cpu/*.cc"
    "strings/*.cc"
    "strings/cpu/*.cc"
    "onednn/*.cc")
else()
  file(
    GLOB
    kernel_cc
    "*.cc"
    "cpu/*.cc"
    "selected_rows/*.cc"
    "selected_rows/cpu/*.cc"
    "sparse/*.cc"
    "sparse/cpu/*.cc"
    "strings/*.cc"
    "strings/cpu/*.cc")
endif()

128
file(GLOB kernel_xpu "xpu/*.cc" "selected_rows/xpu/*.cc")
129 130 131 132

add_library(phi_cpu ${kernel_cc})
kernel_declare("${kernel_cc}")
target_link_libraries(phi_cpu ${COMMON_KERNEL_DEPS})
133 134

set(ADD_PHI_KERNELS phi_cpu)
135 136 137 138 139 140 141 142 143

if(WITH_GPU OR WITH_ROCM)
  if(WITH_GPU)
    add_library(phi_gpu ${kernel_cu})
  elseif(WITH_ROCM)
    hip_add_library(phi_gpu STATIC ${kernel_cu})
  endif()
  kernel_declare("${kernel_cu}")
  target_link_libraries(phi_gpu ${COMMON_KERNEL_DEPS})
144
  set(ADD_PHI_KERNELS ${ADD_PHI_KERNELS} phi_gpu)
145 146 147 148
endif()

if(WITH_XPU)
  if(WITH_XPU_KP)
L
Leo Chen 已提交
149 150 151 152 153 154 155 156 157 158
    file(COPY ${CMAKE_CURRENT_SOURCE_DIR}/kps/
         DESTINATION ${CMAKE_CURRENT_BINARY_DIR}/kps/)
    file(GLOB kernel_xpu_kps "${CMAKE_CURRENT_BINARY_DIR}/kps/*.cu")
    foreach(kernel ${kernel_xpu_kps})
      get_filename_component(name ${kernel} NAME_WE)
      file(RENAME ${kernel} "${CMAKE_CURRENT_BINARY_DIR}/kps/${name}.kps")
    endforeach()
    file(GLOB kernel_xpu_kps "${CMAKE_CURRENT_BINARY_DIR}/kps/*.kps")
    xpu_add_library(phi_xpu STATIC ${kernel_xpu} ${kernel_xpu_kps} DEPENDS
                    ${COMMON_KERNEL_DEPS})
159 160 161
  else()
    add_library(phi_xpu ${kernel_xpu})
  endif()
162 163
  kernel_declare("${kernel_xpu}")
  kernel_declare("${kernel_xpu_kps}")
164
  target_link_libraries(phi_xpu ${COMMON_KERNEL_DEPS})
165 166 167 168
  set(ADD_PHI_KERNELS ${ADD_PHI_KERNELS} phi_xpu)
endif()

set_property(GLOBAL PROPERTY PHI_KERNELS ${ADD_PHI_KERNELS})