/
CMakeLists.txt
208 lines (197 loc) · 6.62 KB
/
CMakeLists.txt
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
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
find_package(Python3 REQUIRED COMPONENTS Interpreter)
macro(pnnx_ncnn_add_test name)
add_test(NAME test_ncnn_${name} COMMAND ${CMAKE_COMMAND} -DPYTHON_EXECUTABLE=${Python3_EXECUTABLE} -DPYTHON_SCRIPT=${CMAKE_CURRENT_SOURCE_DIR}/test_${name}.py -P ${CMAKE_CURRENT_SOURCE_DIR}/../run_test.cmake)
endmacro()
pnnx_ncnn_add_test(F_adaptive_avg_pool1d)
pnnx_ncnn_add_test(F_adaptive_avg_pool2d)
pnnx_ncnn_add_test(F_adaptive_avg_pool3d)
pnnx_ncnn_add_test(F_adaptive_max_pool1d)
pnnx_ncnn_add_test(F_adaptive_max_pool2d)
pnnx_ncnn_add_test(F_adaptive_max_pool3d)
pnnx_ncnn_add_test(F_alpha_dropout)
pnnx_ncnn_add_test(F_avg_pool1d)
pnnx_ncnn_add_test(F_avg_pool2d)
pnnx_ncnn_add_test(F_avg_pool3d)
pnnx_ncnn_add_test(F_batch_norm)
pnnx_ncnn_add_test(F_conv_transpose1d)
pnnx_ncnn_add_test(F_conv_transpose2d)
pnnx_ncnn_add_test(F_conv_transpose3d)
pnnx_ncnn_add_test(F_conv1d)
pnnx_ncnn_add_test(F_conv2d)
pnnx_ncnn_add_test(F_conv3d)
pnnx_ncnn_add_test(F_dropout)
pnnx_ncnn_add_test(F_dropout2d)
pnnx_ncnn_add_test(F_dropout3d)
pnnx_ncnn_add_test(F_elu)
pnnx_ncnn_add_test(F_embedding)
pnnx_ncnn_add_test(F_feature_alpha_dropout)
pnnx_ncnn_add_test(F_fold)
pnnx_ncnn_add_test(F_gelu)
pnnx_ncnn_add_test(F_glu)
pnnx_ncnn_add_test(F_grid_sample)
pnnx_ncnn_add_test(F_group_norm)
pnnx_ncnn_add_test(F_hardsigmoid)
pnnx_ncnn_add_test(F_hardswish)
pnnx_ncnn_add_test(F_hardtanh)
pnnx_ncnn_add_test(F_interpolate)
pnnx_ncnn_add_test(F_layer_norm)
pnnx_ncnn_add_test(F_leaky_relu)
pnnx_ncnn_add_test(F_local_response_norm)
pnnx_ncnn_add_test(F_max_pool1d)
pnnx_ncnn_add_test(F_max_pool2d)
pnnx_ncnn_add_test(F_max_pool3d)
pnnx_ncnn_add_test(F_normalize)
pnnx_ncnn_add_test(F_pad)
pnnx_ncnn_add_test(F_pixel_shuffle)
pnnx_ncnn_add_test(F_pixel_unshuffle)
pnnx_ncnn_add_test(F_prelu)
pnnx_ncnn_add_test(F_relu)
pnnx_ncnn_add_test(F_relu6)
pnnx_ncnn_add_test(F_sigmoid)
pnnx_ncnn_add_test(F_silu)
pnnx_ncnn_add_test(F_softmax)
pnnx_ncnn_add_test(F_tanh)
pnnx_ncnn_add_test(F_unfold)
pnnx_ncnn_add_test(F_upsample_bilinear)
pnnx_ncnn_add_test(F_upsample_nearest)
pnnx_ncnn_add_test(F_upsample)
pnnx_ncnn_add_test(nn_AdaptiveAvgPool1d)
pnnx_ncnn_add_test(nn_AdaptiveAvgPool2d)
pnnx_ncnn_add_test(nn_AdaptiveAvgPool3d)
pnnx_ncnn_add_test(nn_AdaptiveMaxPool1d)
pnnx_ncnn_add_test(nn_AdaptiveMaxPool2d)
pnnx_ncnn_add_test(nn_AdaptiveMaxPool3d)
pnnx_ncnn_add_test(nn_AlphaDropout)
pnnx_ncnn_add_test(nn_AvgPool1d)
pnnx_ncnn_add_test(nn_AvgPool2d)
pnnx_ncnn_add_test(nn_AvgPool3d)
pnnx_ncnn_add_test(nn_BatchNorm1d)
pnnx_ncnn_add_test(nn_BatchNorm2d)
pnnx_ncnn_add_test(nn_BatchNorm3d)
pnnx_ncnn_add_test(nn_ChannelShuffle)
pnnx_ncnn_add_test(nn_ConstantPad1d)
pnnx_ncnn_add_test(nn_ConstantPad2d)
pnnx_ncnn_add_test(nn_ConstantPad3d)
pnnx_ncnn_add_test(nn_Conv1d)
pnnx_ncnn_add_test(nn_Conv2d)
pnnx_ncnn_add_test(nn_Conv3d)
pnnx_ncnn_add_test(nn_ConvTranspose1d)
pnnx_ncnn_add_test(nn_ConvTranspose2d)
pnnx_ncnn_add_test(nn_ConvTranspose3d)
pnnx_ncnn_add_test(nn_Dropout)
pnnx_ncnn_add_test(nn_Dropout2d)
pnnx_ncnn_add_test(nn_Dropout3d)
pnnx_ncnn_add_test(nn_ELU)
pnnx_ncnn_add_test(nn_Embedding)
pnnx_ncnn_add_test(nn_Fold)
pnnx_ncnn_add_test(nn_GELU)
pnnx_ncnn_add_test(nn_GLU)
pnnx_ncnn_add_test(nn_GroupNorm)
pnnx_ncnn_add_test(nn_GRU)
pnnx_ncnn_add_test(nn_Hardsigmoid)
pnnx_ncnn_add_test(nn_Hardswish)
pnnx_ncnn_add_test(nn_Hardtanh)
pnnx_ncnn_add_test(nn_InstanceNorm2d)
pnnx_ncnn_add_test(nn_LayerNorm)
pnnx_ncnn_add_test(nn_LeakyReLU)
pnnx_ncnn_add_test(nn_Linear)
pnnx_ncnn_add_test(nn_LocalResponseNorm)
pnnx_ncnn_add_test(nn_LSTM)
pnnx_ncnn_add_test(nn_MaxPool1d)
pnnx_ncnn_add_test(nn_MaxPool2d)
pnnx_ncnn_add_test(nn_MaxPool3d)
pnnx_ncnn_add_test(nn_MultiheadAttention)
pnnx_ncnn_add_test(nn_PixelShuffle)
pnnx_ncnn_add_test(nn_PixelUnshuffle)
pnnx_ncnn_add_test(nn_PReLU)
pnnx_ncnn_add_test(nn_ReflectionPad1d)
pnnx_ncnn_add_test(nn_ReflectionPad2d)
pnnx_ncnn_add_test(nn_ReLU)
pnnx_ncnn_add_test(nn_ReLU6)
pnnx_ncnn_add_test(nn_ReplicationPad1d)
pnnx_ncnn_add_test(nn_ReplicationPad2d)
pnnx_ncnn_add_test(nn_RNN)
pnnx_ncnn_add_test(nn_SELU)
pnnx_ncnn_add_test(nn_Sigmoid)
pnnx_ncnn_add_test(nn_SiLU)
pnnx_ncnn_add_test(nn_Softmax)
pnnx_ncnn_add_test(nn_Softmax2d)
pnnx_ncnn_add_test(nn_Tanh)
pnnx_ncnn_add_test(nn_Unfold)
pnnx_ncnn_add_test(nn_Upsample)
pnnx_ncnn_add_test(nn_UpsamplingBilinear2d)
pnnx_ncnn_add_test(nn_UpsamplingNearest2d)
pnnx_ncnn_add_test(nn_ZeroPad2d)
pnnx_ncnn_add_test(Tensor_contiguous)
pnnx_ncnn_add_test(Tensor_repeat)
pnnx_ncnn_add_test(Tensor_reshape)
pnnx_ncnn_add_test(Tensor_slice)
pnnx_ncnn_add_test(Tensor_slice_copy)
pnnx_ncnn_add_test(Tensor_view)
pnnx_ncnn_add_test(torch_addmm)
pnnx_ncnn_add_test(torch_amax)
pnnx_ncnn_add_test(torch_amin)
pnnx_ncnn_add_test(torch_bmm)
pnnx_ncnn_add_test(torch_cat)
pnnx_ncnn_add_test(torch_chunk)
pnnx_ncnn_add_test(torch_clone)
pnnx_ncnn_add_test(torch_cumsum)
pnnx_ncnn_add_test(torch_einsum)
pnnx_ncnn_add_test(torch_logsumexp)
pnnx_ncnn_add_test(torch_matmul)
pnnx_ncnn_add_test(torch_max)
pnnx_ncnn_add_test(torch_mean)
pnnx_ncnn_add_test(torch_min)
pnnx_ncnn_add_test(torch_mm)
pnnx_ncnn_add_test(torch_norm)
pnnx_ncnn_add_test(torch_permute)
pnnx_ncnn_add_test(torch_prod)
pnnx_ncnn_add_test(torch_sum)
pnnx_ncnn_add_test(torch_t)
pnnx_ncnn_add_test(torch_squeeze)
pnnx_ncnn_add_test(torch_stack)
pnnx_ncnn_add_test(torch_tensor_split)
pnnx_ncnn_add_test(torch_transpose)
pnnx_ncnn_add_test(torch_unbind)
pnnx_ncnn_add_test(torch_unsqueeze)
pnnx_ncnn_add_test(torch_abs)
pnnx_ncnn_add_test(torch_acos)
pnnx_ncnn_add_test(torch_asin)
pnnx_ncnn_add_test(torch_atan)
pnnx_ncnn_add_test(torch_atan2)
pnnx_ncnn_add_test(torch_ceil)
pnnx_ncnn_add_test(torch_clamp)
pnnx_ncnn_add_test(torch_cos)
pnnx_ncnn_add_test(torch_exp)
pnnx_ncnn_add_test(torch_floor)
pnnx_ncnn_add_test(torch_log)
pnnx_ncnn_add_test(torch_log10)
pnnx_ncnn_add_test(torch_neg)
pnnx_ncnn_add_test(torch_pow)
pnnx_ncnn_add_test(torch_reciprocal)
pnnx_ncnn_add_test(torch_round)
pnnx_ncnn_add_test(torch_rsqrt)
pnnx_ncnn_add_test(torch_sin)
pnnx_ncnn_add_test(torch_sqrt)
pnnx_ncnn_add_test(torch_square)
pnnx_ncnn_add_test(torch_tan)
pnnx_ncnn_add_test(torch_tanh)
pnnx_ncnn_add_test(torch_trunc)
pnnx_ncnn_add_test(convnext_tiny)
pnnx_ncnn_add_test(mobilenet_v2)
pnnx_ncnn_add_test(mobilenet_v3_small)
pnnx_ncnn_add_test(resnet18)
pnnx_ncnn_add_test(shufflenet_v2_x1_0)
pnnx_ncnn_add_test(squeezenet1_1)
pnnx_ncnn_add_test(vit_b_32)
pnnx_ncnn_add_test(ncnn_fuse_transpose_matmul)
pnnx_ncnn_add_test(ncnn_fuse_shufflechannel_slice)
pnnx_ncnn_add_test(ncnn_fuse_binaryop_eltwise)
pnnx_ncnn_add_test(ncnn_numpy_binaryop_broadcast)
if(Torch_VERSION VERSION_GREATER_EQUAL "1.9")
pnnx_ncnn_add_test(F_mish)
pnnx_ncnn_add_test(nn_Mish)
endif()
if(TorchVision_FOUND)
pnnx_ncnn_add_test(torchvision_DeformConv2d)
endif()