improve test of general shape_cast reshape

Author: newling

mlir/test/Dialect/Vector/ops.mlir

@@ -335,7 +335,7 @@ func.func @outerproduct_scalable(%arg0 : vector<[4]xf32>, %arg1 : vector<[8]xf32
 
 // CHECK-LABEL: @insert_strided_slice
 func.func @insert_strided_slice(%a: vector<4x4xf32>, %b: vector<4x8x16xf32>) {
-  // CHECK: vector.insert_strided_slice %{{.*}}, %{{.*}} {offsets = [2, 2, 2], strides = [1, 1]} : vector<4x4xf32> into vector<4x8x16xf32>
+  // CHECK: vector.insert_strided_slice %{{.*}}, %{{.*}} { ffsets = [2, 2, 2], strides = [1, 1]} : vector<4x4xf32> into vector<4x8x16xf32>
   %1 = vector.insert_strided_slice %a, %b {offsets = [2, 2, 2], strides = [1, 1]} : vector<4x4xf32> into vector<4x8x16xf32>
   return
 }
@@ -543,20 +543,6 @@ func.func @vector_print_on_scalar(%arg0: i64) {
   return
 }
 
-// CHECK-LABEL: @shape_cast_valid_rank_reduction
-func.func @shape_cast_valid_rank_reduction(%arg0 : vector<5x1x3x2xf32>) {
-  // CHECK: vector.shape_cast %{{.*}} : vector<5x1x3x2xf32> to vector<2x15xf32>
-  %0 = vector.shape_cast %arg0 : vector<5x1x3x2xf32> to vector<2x15xf32>
-  return
-}
-
-// CHECK-LABEL: @shape_cast_valid_rank_expansion
-func.func @shape_cast_valid_rank_expansion(%arg0 : vector<15x2xf32>) {
-  // CHECK: vector.shape_cast %{{.*}} : vector<15x2xf32> to vector<5x2x3x1xf32>
-  %0 = vector.shape_cast %arg0 : vector<15x2xf32> to vector<5x2x3x1xf32>
-  return
-}
-
 // CHECK-LABEL: @shape_cast
 func.func @shape_cast(%arg0 : vector<5x1x3x2xf32>,
                  %arg1 : vector<8x1xf32>,
@@ -578,6 +564,17 @@ func.func @shape_cast(%arg0 : vector<5x1x3x2xf32>,
   return %0, %1, %2, %3 : vector<15x2xf32>, vector<8xf32>, vector<16xf32>, vector<16x1xf32>
 }
 
+// A vector.shape_cast can cast between any 2 shapes as long as the
+// number of elements is preserved. For those familiar with the tensor
+// dialect: this behaviour is like the tensor.reshape operation, i.e.
+// less restrictive than tensor.collapse_shape and tensor.expand_shape
+// CHECK-LABEL: @shape_cast_general_reshape
+func.func @shape_cast_general_reshape(%arg0 : vector<2x3xf32>) -> (vector<3x1x2xf32>) {
+  // CHECK: vector.shape_cast %{{.*}} : vector<2x3xf32> to vector<3x1x2xf32>
+  %0 = vector.shape_cast %arg0 : vector<2x3xf32> to vector<3x1x2xf32>
+  return %0
+}
+
 // CHECK-LABEL: @shape_cast_0d
 func.func @shape_cast_0d(%arg0 : vector<1x1x1x1xf32>) -> (vector<1x1x1x1xf32>) {
 

mlir/test/Dialect/Vector/playtime.mlir

@@ -0,0 +1,14 @@
+func.func @test_extract_strided_slice_1(%arg0 : vector<4x4x4x4xf32>) -> vector<2x3x4x4xf32> {
+  %0 = vector.extract_strided_slice %arg0 { sizes = [2, 3], strides = [1, 1], offsets = [1, 1]}
+     : vector<4x4x4x4xf32> to vector<2x3x4x4xf32>
+  return %0 : vector<2x3x4x4xf32>
+}
+
+
+func.func @test(%arg0 : vector<7xf32>, %arg1 : vector<3xf32>) -> vector<4xf32> {
+
+  %out = vector.shuffle %arg0, %arg1 [0, 4, 5, 1]
+    : vector<7xf32>, vector<3xf32> 
+    return %out : vector<4xf32>
+
+}