better grouping of tests

Author: newling

mlir/test/Dialect/Vector/canonicalize/vector-transpose.mlir

@@ -1,6 +1,10 @@
 // RUN: mlir-opt %s -canonicalize="test-convergence" -split-input-file -allow-unregistered-dialect | FileCheck %s
 
-// This file contains some canonicalizations tests involving vector.transpose.
+// This file contains some tests of canonicalizations and foldings involving vector.transpose.
+
+// +----------------------------------------
+//  Tests of FoldTransposeBroadcast
+// +----------------------------------------
 
 // CHECK-LABEL: func @transpose_scalar_broadcast1
 //  CHECK-SAME: (%[[ARG:.+]]: vector<1xf32>)
@@ -251,7 +255,10 @@ func.func @negative_transpose_of_shape_cast(%arg : vector<6xi8>) -> vector<2x3xi
 
 // -----
 
-// Test of transpose folding
+// +-----------------------------------
+//  Tests of transpose folding
+// +-----------------------------------
+
 //  CHECK-LABEL: transpose_1D_identity
 //   CHECK-SAME:   [[ARG:%.*]]: vector<4xf32>
 //   CHECK-NEXT:   return [[ARG]]
@@ -262,7 +269,6 @@ func.func @transpose_1D_identity(%arg : vector<4xf32>) -> vector<4xf32> {
 
 // -----
 
-// Test of transpose folding
 // CHECK-LABEL: transpose_2D_identity
 //  CHECK-SAME:    [[ARG:%.*]]: vector<4x3xf32>
 //  CHECK-NEXT:    return [[ARG]]
@@ -273,11 +279,20 @@ func.func @transpose_2D_identity(%arg : vector<4x3xf32>) -> vector<4x3xf32> {
 
 // -----
 
-// Test of transpose folding
 // CHECK-LABEL: transpose_shape_and_order_preserving
 //  CHECK-SAME:    [[ARG:%.*]]: vector<6x1x1x4xi8>
 //  CHECK-NEXT:    return [[ARG]]
 func.func @transpose_shape_and_order_preserving(%arg : vector<6x1x1x4xi8>) -> vector<6x1x1x4xi8> {
   %0 = vector.transpose %arg, [0, 2, 1, 3] : vector<6x1x1x4xi8> to vector<6x1x1x4xi8>
   return %0 : vector<6x1x1x4xi8>
 }
+
+// -----
+
+// CHECK-LABEL: negative_transpose_fold
+//       CHECK: [[TRANSP:%.*]] = vector.transpose
+//       CHECK: return [[TRANSP]]
+func.func @negative_transpose_fold(%arg : vector<2x2xi8>) -> vector<2x2xi8> {
+  %0 = vector.transpose %arg, [1, 0] : vector<2x2xi8> to vector<2x2xi8>
+  return %0 : vector<2x2xi8>
+}