CMakeLists.txt 2.2 KB
Newer Older
1 2 3 4 5
if(WITH_ASCEND_CL)
  cc_library(
    beam_search_npu
    SRCS beam_search_npu.cc
    DEPS npu_op_runner)
P
pangyoki 已提交
6 7
endif()

8 9 10 11 12 13 14
if(WITH_XPU)
  cc_library(
    beam_search_xpu
    SRCS beam_search_xpu.cc
    DEPS xpu_op_list)
endif()

15
# please add new math_library in alphabetical order
16 17 18 19
if(WITH_ASCEND_CL)
  math_library(concat_and_split DEPS concat_and_split_functor npu_op_runner)
elseif(WITH_MLU)
  math_library(concat_and_split DEPS concat_and_split_functor mlu_baseop)
20
else()
21
  math_library(concat_and_split DEPS concat_and_split_functor)
22
endif()
23 24
math_library(context_project DEPS im2col math_function)
math_library(cos_sim_functor)
Z
Zeng Jinle 已提交
25
math_library(depthwise_conv)
X
xuezhong 已提交
26
math_library(sample_prob)
27
math_library(sampler DEPS generator)
D
dzhwinter 已提交
28

F
Feiyu Chan 已提交
29
# math_library(math_function DEPS blas dense_tensor tensor)
L
lidanqing 已提交
30

31
math_library(sequence_padding)
T
tensor-tang 已提交
32
math_library(sequence_pooling DEPS math_function jit_kernel_helper)
33 34
if(WITH_ASCEND_CL)
  math_library(beam_search DEPS math_function beam_search_npu)
35 36
elseif(WITH_XPU)
  math_library(beam_search DEPS math_function beam_search_xpu)
P
pangyoki 已提交
37
else()
38
  math_library(beam_search DEPS math_function)
P
pangyoki 已提交
39
endif()
W
wopeizl 已提交
40

41
math_library(unpooling)
N
nhzlx 已提交
42
math_library(prelu)
43
math_library(bert_encoder_functor)
Z
zhaozhehao 已提交
44
math_library(tree2col DEPS math_function)
Q
qijun 已提交
45

46 47 48
cc_test(
  selected_rows_functor_test
  SRCS selected_rows_functor_test.cc
49
  DEPS allocator selected_rows_functor)
50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69
cc_test(
  im2col_test
  SRCS im2col_test.cc
  DEPS im2col)
cc_test(
  vol2col_test
  SRCS vol2col_test.cc
  DEPS vol2col)
cc_test(
  sequence_padding_test
  SRCS sequence_padding_test.cc
  DEPS sequence_padding)
cc_test(
  sequence_pooling_test
  SRCS sequence_pooling_test.cc
  DEPS sequence_pooling)
cc_test(
  beam_search_test
  SRCS beam_search_test.cc
  DEPS beam_search)
70
if(WITH_GPU)
71 72 73 74
  nv_test(
    selected_rows_functor_gpu_test
    SRCS selected_rows_functor_test.cu.cc
    DEPS selected_rows_functor math_function)
75
endif()
76
if(WITH_ROCM)
77 78 79 80
  hip_test(
    selected_rows_functor_gpu_test
    SRCS selected_rows_functor_test.cu.cc
    DEPS selected_rows_functor math_function)
81
endif()
82 83 84 85
cc_test(
  concat_test
  SRCS concat_test.cc
  DEPS concat_and_split)
Z
zhangkaihuo 已提交
86

87
if(WITH_TESTING AND TEST im2col_test)
88
  set_tests_properties(im2col_test PROPERTIES TIMEOUT 120)
Y
YUNSHEN XIE 已提交
89
endif()