Add a negative test and also update the diagnostic test with lesser ops

Author: Prakhar-Dixit

mlir/test/Dialect/Vector/invalid.mlir

@@ -768,11 +768,10 @@ func.func @extract_strided_slice(%arg0: vector<4x8x16xf32>) {
 
 // -----
 
- func.func @extract_strided_slice() -> () {
+func.func @extract_strided_slice(%arg0: vector<3x2x2xf32>) {
   // expected-error@+1 {{expected input vector rank to match target shape rank}}
-  %0 = arith.constant dense<1.000000e+00> : vector<24x2x2xf32>
-  %1 = vector.extract_strided_slice %0 {offsets = [0, 0], sizes = [2, 2], strides = [1, 1]}:
-         vector<24x2x2xf32> to vector<2x2xf32>
+  %1 = vector.extract_strided_slice %arg0 {offsets = [0, 0], sizes = [2, 2], strides = [1, 1]}:
+         vector<3x2x2xf32> to vector<2x2xf32>
   return
 }
 

mlir/test/Dialect/Vector/vector-unroll-options.mlir

@@ -188,6 +188,20 @@ func.func @vector_fma(%a: vector<4x4xf32>, %b: vector<4x4xf32>, %c: vector<4x4xf
 //   CHECK-LABEL: func @vector_fma
 // CHECK-COUNT-4: vector.fma %{{.+}}, %{{.+}}, %{{.+}} : vector<2x2xf32>
 
+func.func @higher_rank_unroll() {
+  %cst_25 = arith.constant dense<3.718400e+04> : vector<4x2x2xf16>
+  %cst_26 = arith.constant dense<1.000000e+00> : vector<24x2x2xf32>
+  %47 = vector.fma %cst_26, %cst_26, %cst_26 : vector<24x2x2xf32>
+  %818 = scf.execute_region -> vector<24x2x2xf32> {
+      scf.yield %47 : vector<24x2x2xf32>
+    }
+  %823 = vector.extract_strided_slice %cst_25 {offsets = [2], sizes = [1], strides = [1]} : vector<4x2x2xf16> to vector<1x2x2xf16>
+  return
+}
+
+// CHECK-LABEL: func @higher_rank_unroll
+//       CHECK: return 
+
 func.func @vector_multi_reduction(%v : vector<4x6xf32>, %acc: vector<4xf32>) -> vector<4xf32> {
   %0 = vector.multi_reduction #vector.kind<add>, %v, %acc [1] : vector<4x6xf32> to vector<4xf32>
   return %0 : vector<4xf32>