Paddle/test/cpp/phi/kernels/CMakeLists.txt at 72573e2aa00a1ddf51c03961e174b3baacff1268 · youge325/Paddle · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
cc_test(
  test_math_function
  SRCS test_math_function.cc
  DEPS phi common)
if(WITH_GPU)
  nv_test(
    test_math_function_gpu
    SRCS test_math_function.cu
    DEPS phi common)
  nv_test(
    test_broadcast_gpu
    SRCS test_ternary_broadcast.cu
    DEPS gtest)
endif()
if(WITH_ROCM)
  hip_test(
    test_math_function_gpu
    SRCS test_math_function.cu
    DEPS phi common)
endif()

cc_test(
  test_cpu_vec
  SRCS test_cpu_vec.cc
  DEPS phi common)

# For String Kernels
if(WIN32)
  cc_test(
    test_strings_lower_upper_dev_api
    SRCS test_strings_lower_upper_dev_api.cc
    DEPS type_info common)
else()
  cc_test(
    test_strings_lower_upper_dev_api
    SRCS test_strings_lower_upper_dev_api.cc
    DEPS phi common)
endif()
if(WITH_GPU)
  nv_test(
    test_strings_lower_upper_dev_gpu_api
    SRCS test_strings_lower_upper_dev_api.cu
    DEPS phi common)
elseif(WITH_ROCM)
  hip_test(
    test_strings_lower_upper_dev_gpu_api
    SRCS test_strings_lower_upper_dev_api.cu
    DEPS phi common)
endif()

cc_test(
  test_strings_copy_dev_api
  SRCS test_strings_copy_dev_api.cc
  DEPS phi common)
if(WITH_GPU)
  nv_test(
    test_strings_copy_dev_gpu_api
    SRCS test_strings_copy_dev_api.cu
    DEPS phi common)
elseif(WITH_ROCM)
  hip_test(
    test_strings_copy_dev_gpu_api
    SRCS test_strings_copy_dev_api.cu
    DEPS phi common)
endif()

if(WIN32)
  cc_test(
    test_memcpy_dev_api
    SRCS test_memcpy_dev_api.cc
    DEPS type_info common)
  cc_test(
    test_transfer_layout_dev_api
    SRCS test_memcpy_dev_api.cc
    DEPS type_info common)
else()
  cc_test(
    test_memcpy_dev_api
    SRCS test_memcpy_dev_api.cc
    DEPS phi common)
  cc_test(
    test_transfer_layout_dev_api
    SRCS test_transfer_layout_dev_api.cc
    DEPS phi common)
endif()

if(WITH_GPU)
  nv_test(
    test_gpu_timer
    SRCS test_gpu_timer.cu
    DEPS gtest)
  nv_test(
    test_auto_tune
    SRCS test_auto_tune.cu
    DEPS gtest)
  cc_test(
    test_fused_adam_kernel
    SRCS test_fused_adam_kernel.cc
    DEPS gtest phi common)
elseif(WITH_ROCM)
  hip_test(
    test_gpu_timer
    SRCS test_gpu_timer.cu
    DEPS gtest)
  hip_test(
    test_auto_tune
    SRCS test_auto_tune.cu
    DEPS gtest)
endif()

cc_test(
  test_cache
  SRCS test_cache.cc
  DEPS gtest phi common)

cc_test(
  strided_memcpy_test
  SRCS strided_memcpy_test.cc
  DEPS phi common)

cc_test(
  test_isfinite_kernel
  SRCS test_isfinite_kernel.cc
  DEPS gtest phi common)

if(WIN32)
  cc_test(
    sequence_padding_test
    SRCS sequence_padding_test.cc
    DEPS type_info common)
else()
  cc_test(
    sequence_padding_test
    SRCS sequence_padding_test.cc
    DEPS phi common)
endif()

cc_test(
  sequence_pooling_test
  SRCS sequence_pooling_test.cc
  DEPS phi common)