Address syntax changes in test files

aqjune · Nov 13, 2024 · ca5123f · ca5123f
1 parent 2e53307
commit ca5123f
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/tests/litmus/tosa-ops/reduce_sum_int-bad.tgt.mlir b/tests/litmus/tosa-ops/reduce_sum_int-bad.tgt.mlir
@@ -1,5 +1,5 @@
 func.func @f(%t: tensor<3x3x3xi32>) -> tensor<3x3xi32> {
-  %0 = "tosa.reduce_sum"(%t) {axis = 1 : i64} : (tensor<3x3x3xi32>) -> tensor<3x1x3xi32>
+  %0 = "tosa.reduce_sum"(%t) {axis = 1 : i32} : (tensor<3x3x3xi32>) -> tensor<3x1x3xi32>
   %res = tensor.collapse_shape %0 [[0, 1], [2]] : tensor<3x1x3xi32> into tensor<3x3xi32>
 	return %res: tensor<3x3xi32>
 }
diff --git a/tests/opts/tosa-to-linalg/reduce_sum.tgt.mlir b/tests/opts/tosa-to-linalg/reduce_sum.tgt.mlir
@@ -10,7 +10,7 @@ module  {
       %4 = arith.addf %arg1, %arg2 : f32
       linalg.yield %4 : f32
     } -> tensor<4x5xf32>
-    %3 = tensor.expand_shape %2 [[0, 1], [2]] : tensor<4x5xf32> into tensor<1x4x5xf32>
+    %3 = tensor.expand_shape %2 [[0, 1], [2]] output_shape [1,4,5] : tensor<4x5xf32> into tensor<1x4x5xf32>
     return %3 : tensor<1x4x5xf32>
   }
 }

diff --git a/tests/opts/tosa-to-linalg/reduce_sum2.tgt.mlir b/tests/opts/tosa-to-linalg/reduce_sum2.tgt.mlir
@@ -10,7 +10,7 @@ module  {
       %4 = arith.addf %arg1, %arg2 : f32
       linalg.yield %4 : f32
     } -> tensor<3x5xf32>
-    %3 = tensor.expand_shape %2 [[0, 1], [2]] : tensor<3x5xf32> into tensor<3x1x5xf32>
+    %3 = tensor.expand_shape %2 [[0, 1], [2]] output_shape [3,1,5] : tensor<3x5xf32> into tensor<3x1x5xf32>
     return %3 : tensor<3x1x5xf32>
   }
 }

diff --git a/tests/opts/tosa-to-linalg/reduce_sum3.tgt.mlir b/tests/opts/tosa-to-linalg/reduce_sum3.tgt.mlir
@@ -10,7 +10,7 @@ module  {
       %4 = arith.addf %arg1, %arg2 : f32
       linalg.yield %4 : f32
     } -> tensor<3x5xf32>
-    %3 = tensor.expand_shape %2 [[0, 1], [2]] : tensor<3x5xf32> into tensor<3x1x5xf32>
+    %3 = tensor.expand_shape %2 [[0, 1], [2]] output_shape [3,1,5] : tensor<3x5xf32> into tensor<3x1x5xf32>
     return %3 : tensor<3x1x5xf32>
   }
 }