enable tests

Tencent · Nov 21, 2024 · 8ab9362 · 8ab9362
1 parent f16caff
commit 8ab9362
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/tools/pnnx/tests/CMakeLists.txt b/tools/pnnx/tests/CMakeLists.txt
@@ -360,6 +360,11 @@ if(TorchVision_FOUND)
     pnnx_add_test(torchvision_RoIAlign)
 endif()
 
+pnnx_add_test(torchaudio_F_inverse_spectrogram)
+pnnx_add_test(torchaudio_F_spectrogram)
+pnnx_add_test(torchaudio_InverseSpectrogram)
+pnnx_add_test(torchaudio_Spectrogram)
+
 add_subdirectory(ncnn)
 
 if(onnxruntime_FOUND)

diff --git a/tools/pnnx/tests/ncnn/CMakeLists.txt b/tools/pnnx/tests/ncnn/CMakeLists.txt
@@ -175,6 +175,9 @@ pnnx_ncnn_add_test(torch_transpose)
 pnnx_ncnn_add_test(torch_unbind)
 pnnx_ncnn_add_test(torch_unsqueeze)
 
+pnnx_ncnn_add_test(torch_istft)
+pnnx_ncnn_add_test(torch_stft)
+
 pnnx_ncnn_add_test(torch_abs)
 pnnx_ncnn_add_test(torch_acos)
 pnnx_ncnn_add_test(torch_asin)
@@ -217,3 +220,8 @@ pnnx_ncnn_add_test(ncnn_numpy_binaryop_broadcast)
 if(TorchVision_FOUND)
     pnnx_ncnn_add_test(torchvision_DeformConv2d)
 endif()
+
+pnnx_ncnn_add_test(torchaudio_F_inverse_spectrogram)
+pnnx_ncnn_add_test(torchaudio_F_spectrogram)
+pnnx_ncnn_add_test(torchaudio_InverseSpectrogram)
+pnnx_ncnn_add_test(torchaudio_Spectrogram)