added test

xdslproject · Oct 15, 2024 · 0410875 · 0410875
1 parent 1418998
commit 0410875
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/filecheck/mlir-conversion/with-mlir/dialects/tensor/ops.mlir b/tests/filecheck/mlir-conversion/with-mlir/dialects/tensor/ops.mlir
@@ -13,6 +13,8 @@
 %dim2 = "tensor.dim"(%t1, %i1) {"hello" = "world"}: (tensor<2x3xf32>, index) -> index
 %cast1 = "tensor.cast"(%t1) : (tensor<2x3xf32>) -> tensor<?x?xf32>
 %cast2 = "tensor.cast"(%t1) {"hello" = "world"} : (tensor<2x3xf32>) -> tensor<?x?xf32>
+%big_tensor = tensor.empty() : tensor<2x3x2x3xf32>
+%collapsed = tensor.collapse_shape %big_tensor [[0, 1], [2, 3]] : tensor<2x3x2x3xf32> into tensor<6x6xf32>
 
 
 // CHECK:       module {
@@ -29,4 +31,6 @@
 // CHECK-NEXT:  %{{.*}} = tensor.dim {hello = "world"} %{{.*}}, %{{.*}} : tensor<2x3xf32>
 // CHECK-NEXT:  %{{.*}} = tensor.cast %{{.*}} : tensor<2x3xf32> to tensor<?x?xf32>
 // CHECK-NEXT:  %{{.*}} = tensor.cast %{{.*}} {hello = "world"} : tensor<2x3xf32> to tensor<?x?xf32>
+// CHECK-NEXT:  %6 = tensor.empty() : tensor<2x3x2x3xf32>
+// CHECK-NEXT:  %collapsed = tensor.collapse_shape %6 [[0, 1], [2, 3]] : tensor<2x3x2x3xf32> into tensor<6x6xf32>
 // CHECK-NEXT: }