From 157bd6697d39e965e0df4f44e5cc2b0d48fa695b Mon Sep 17 00:00:00 2001 From: Liqun Fu Date: Sat, 25 Nov 2023 16:10:06 -0800 Subject: [PATCH] extended test cases to axis=0 for since version being 20. Signed-off-by: Liqun Fu --- onnxruntime/test/providers/cpu/signal/signal_ops_test.cc | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/onnxruntime/test/providers/cpu/signal/signal_ops_test.cc b/onnxruntime/test/providers/cpu/signal/signal_ops_test.cc index f6690b9ef4815..54d725defe5ee 100644 --- a/onnxruntime/test/providers/cpu/signal/signal_ops_test.cc +++ b/onnxruntime/test/providers/cpu/signal/signal_ops_test.cc @@ -142,9 +142,11 @@ static void TestDFTInvertible(bool complex, int since_version) { RandomValueGenerator random(GetTestRandomSeed()); // TODO(smk2007): Add tests for different dft_length values. constexpr int64_t num_batches = 2; - for (int64_t axis = 1; axis < 2; axis += 1) { + for (int64_t axis = 0; axis < 2; axis += 1) { for (int64_t signal_dim1 = 2; signal_dim1 <= 5; signal_dim1 += 1) { for (int64_t signal_dim2 = 2; signal_dim2 <= 5; signal_dim2 += 1) { + if (axis == 0 && since_version < kOpsetVersion20) + continue; DFTInvertibleTester test(axis, since_version); vector input_shape{num_batches, signal_dim1, signal_dim2, 1 + (complex ? 1 : 0)}; vector input_data = random.Uniform(input_shape, -100.f, 100.f);