extended test cases to axis=0 for since version being 20.

Signed-off-by: Liqun Fu <[email protected]>
microsoft · Nov 26, 2023 · 157bd66 · 157bd66
1 parent 9fdafa9
commit 157bd66
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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<int64_t> input_shape{num_batches, signal_dim1, signal_dim2, 1 + (complex ? 1 : 0)};
         vector<float> input_data = random.Uniform<float>(input_shape, -100.f, 100.f);