fix splat testing

Author: newling

mlir/test/Dialect/Vector/linearize.mlir

@@ -321,35 +321,27 @@ func.func @test_vector_bitcast(%arg0: vector<[4]x2xf32>) -> vector<[4]x4xf16> {
 }
 
 // -----
-// ALL-LABEL: linearize_vector_splat
-// ALL-SAME: (%[[ARG:.*]]: i32) -> vector<4x2xi32>
+
+// CHECK-LABEL: linearize_vector_splat
+// CHECK-SAME: (%[[ARG:.*]]: i32) -> vector<4x2xi32>
 func.func @linearize_vector_splat(%arg0: i32) -> vector<4x2xi32> {
-  // DEFAULT: %[[SPLAT:.*]] = vector.splat %[[ARG]] : vector<8xi32>
-  // DEFAULT: %[[CAST:.*]] = vector.shape_cast %[[SPLAT]] : vector<8xi32> to vector<4x2xi32>
-  // DEFAULT: return %[[CAST]] : vector<4x2xi32>
-  // BW-128: %[[SPLAT:.*]] = vector.splat %[[ARG]] : vector<8xi32>
-  // BW-128: %[[CAST:.*]] = vector.shape_cast %[[SPLAT]] : vector<8xi32> to vector<4x2xi32>
-  // BW-128: return %[[CAST]] : vector<4x2xi32>
-
-  // BW-0: %[[SPLAT:.*]] = vector.splat %[[ARG]] : vector<4x2xi32>
-  // BW-0: return %[[SPLAT]] : vector<4x2xi32>
+
+  // CHECK: %[[SPLAT:.*]] = vector.splat %[[ARG]] : vector<8xi32>
+  // CHECK: %[[CAST:.*]] = vector.shape_cast %[[SPLAT]] : vector<8xi32> to vector<4x2xi32>
+  // CHECK: return %[[CAST]] : vector<4x2xi32>
   %0 = vector.splat %arg0 : vector<4x2xi32>
   return %0 : vector<4x2xi32>
 }
 
 // -----
-// ALL-LABEL: linearize_scalable_vector_splat
-// ALL-SAME: (%[[ARG:.*]]: i32) -> vector<4x[2]xi32>
+
+// CHECK-LABEL: linearize_scalable_vector_splat
+// CHECK-SAME: (%[[ARG:.*]]: i32) -> vector<4x[2]xi32>
 func.func @linearize_scalable_vector_splat(%arg0: i32) -> vector<4x[2]xi32> {
-  // DEFAULT: %[[SPLAT:.*]] = vector.splat %[[ARG]] : vector<[8]xi32>
-  // DEFAULT: %[[CAST:.*]] = vector.shape_cast %[[SPLAT]] : vector<[8]xi32> to vector<4x[2]xi32>
-  // DEFAULT: return %[[CAST]] : vector<4x[2]xi32>
-  // BW-128: %[[SPLAT:.*]] = vector.splat %[[ARG]] : vector<[8]xi32>
-  // BW-128: %[[CAST:.*]] = vector.shape_cast %[[SPLAT]] : vector<[8]xi32> to vector<4x[2]xi32>
-  // BW-128: return %[[CAST]] : vector<4x[2]xi32>
-
-  // BW-0: %[[SPLAT:.*]] = vector.splat %[[ARG]] : vector<4x[2]xi32>
-  // BW-0: return %[[SPLAT]] : vector<4x[2]xi32>
+
+  // CHECK: %[[SPLAT:.*]] = vector.splat %[[ARG]] : vector<[8]xi32>
+  // CHECK: %[[CAST:.*]] = vector.shape_cast %[[SPLAT]] : vector<[8]xi32> to vector<4x[2]xi32>
+  // CHECK: return %[[CAST]] : vector<4x[2]xi32>
   %0 = vector.splat %arg0 : vector<4x[2]xi32>
   return %0 : vector<4x[2]xi32>
 }