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>)
@@ -140,7 +144,10 @@ func.func @negative_broadcast_transpose_021(%arg0 : vector<3x1x3xi8>) -> vector<
 
 // -----
 
-// Test of FoldTransposeShapeCast
+// +-------------------------------------------
+//  Tests of FoldTransposeShapeCast
+// +-------------------------------------------
+
 // In this test, the permutation maps the non-unit dimensions (1 and 2) as follows:
 // 1 -> 0
 // 2 -> 4
@@ -159,7 +166,6 @@ func.func @transpose_shape_cast(%arg : vector<1x4x4x1x1xi8>) -> vector<4x4xi8> {
 
 // -----
 
-// Test of FoldTransposeShapeCast
 // In this test, the mapping of non-unit dimensions (1 and 2) is as follows:
 // 1 -> 2
 // 2 -> 1
@@ -179,7 +185,6 @@ func.func @negative_transpose_shape_cast(%arg : vector<1x4x4x1xi8>) -> vector<4x
 
 // -----
 
-// Test of FoldTransposeShapeCast
 // Currently the conversion shape_cast(transpose) -> shape_cast is disabled for
 // scalable vectors because of bad interaction with ConvertIllegalShapeCastOpsToTransposes
 // CHECK-LABEL: @negative_transpose_shape_cast_scalable
@@ -193,7 +198,10 @@ func.func @negative_transpose_shape_cast_scalable(%arg : vector<[4]x1xi8>) -> ve
 
 // -----
 
-// Test of shape_cast folding.
+// +----------------------------------------
+//  Tests of shape_cast folding
+// +----------------------------------------
+
 // The conversion transpose(shape_cast) -> shape_cast is not disabled for scalable
 // vectors.
 // CHECK-LABEL: @shape_cast_transpose_scalable
@@ -207,7 +215,6 @@ func.func @shape_cast_transpose_scalable(%arg : vector<[4]xi8>) -> vector<[4]x1x
 
 // -----
 
-// Test of shape_cast folding.
 // A transpose that is 'order preserving' can be treated like a shape_cast. 
 // CHECK-LABEL: @shape_cast_transpose
 //  CHECK-SAME:   %[[ARG:.*]]: vector<2x3x1x1xi8>) -> vector<6x1x1xi8> {
@@ -223,7 +230,6 @@ func.func @shape_cast_transpose(%arg : vector<2x3x1x1xi8>) ->  vector<6x1x1xi8>
 
 // -----
 
-// Test of shape_cast folding.
 // Scalable dimensions should be treated as non-unit dimensions.
 // CHECK-LABEL: @shape_cast_transpose_scalable
 //       CHECK: vector.shape_cast
@@ -236,7 +242,6 @@ func.func @shape_cast_transpose_scalable_unit(%arg : vector<[1]x4x1xi8>) -> vect
 
 // -----
 
-// Test of shape_cast (not) folding.
 // CHECK-LABEL: @negative_shape_cast_transpose
 //  CHECK-SAME:   %[[ARG:.*]]: vector<6xi8>) -> vector<2x3xi8> {
 //       CHECK:   %[[SHAPE_CAST:.*]] = vector.shape_cast %[[ARG]] :
@@ -250,7 +255,10 @@ func.func @negative_shape_cast_transpose(%arg : vector<6xi8>) -> vector<2x3xi8>
 
 // -----
 
-// Test of transpose folding
+// +-----------------------------------
+//  Tests of transpose folding
+// +-----------------------------------
+
 //  CHECK-LABEL: transpose_1D_identity
 //   CHECK-SAME:   [[ARG:%.*]]: vector<4xf32>
 //   CHECK-NEXT:   return [[ARG]]
@@ -261,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]]
@@ -272,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>
+}