Skip to content

Commit fbbf149

Browse files
committed
Adding some check-labels so the tests won't fail
1 parent 0ea74a7 commit fbbf149

File tree

1 file changed

+14
-1
lines changed

1 file changed

+14
-1
lines changed

compiler/src/iree/compiler/Codegen/Common/test/decompose_memref.mlir

+14-1
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ func.func @load_scalar_from_memref(%input: memref<4x8xf32>) -> f32 {
1212
%value = memref.load %input[%c0, %c1] : memref<4x8xf32>
1313
return %value : f32
1414
}
15+
// CHECK-LABEL: func @load_scalar_from_memref
1516

1617
// -----
1718

@@ -21,13 +22,15 @@ func.func @load_scalar_from_memref_static_dim(%input: memref<4x8xf32, strided<[8
2122
%value = memref.load %input[%c0, %c1] : memref<4x8xf32, strided<[8, 12], offset: 100>>
2223
return %value : f32
2324
}
25+
// CHECK-LABEL: func @load_scalar_from_memref_static_dim
2426

2527
// -----
2628

2729
func.func @load_scalar_from_memref_static_dim_2(%input: memref<4x8xf32, strided<[8, 12], offset: 100>>, %row: index, %col: index) -> f32 {
2830
%value = memref.load %input[%row, %col] : memref<4x8xf32, strided<[8, 12], offset: 100>>
2931
return %value : f32
3032
}
33+
// CHECK-LABEL: func @load_scalar_from_memref_static_dim_2
3134

3235
// -----
3336

@@ -37,34 +40,39 @@ func.func @load_scalar_from_memref_dynamic_dim(%input: memref<4x8xf32, strided<[
3740
%value = memref.load %input[%c0, %c1] : memref<4x8xf32, strided<[?, ?], offset: ?>>
3841
return %value : f32
3942
}
43+
// CHECK-LABEL: func @load_scalar_from_memref_dynamic_dim
4044

4145
// -----
4246

4347
func.func @load_scalar_from_memref_dynamic_dim_2(%input: memref<4x8xf32, strided<[?, ?], offset: ?>>, %row: index, %col: index) -> f32 {
4448
%value = memref.load %input[%row, %col] : memref<4x8xf32, strided<[?, ?], offset: ?>>
4549
return %value : f32
4650
}
51+
// CHECK-LABEL: func @load_scalar_from_memref_dynamic_dim_2
4752

4853
// -----
4954

5055
func.func @load_scalar_from_memref_subview(%input: memref<4x8xf32>, %row: index, %col: index) -> memref<1x1xf32, strided<[8, 1], offset: ?>> {
5156
%subview = memref.subview %input[%row, %col] [1, 1] [1, 1] : memref<4x8xf32> to memref<1x1xf32, strided<[8, 1], offset: ?>>
5257
return %subview : memref<1x1xf32, strided<[8, 1], offset: ?>>
5358
}
59+
// CHECK-LABEL: func @load_scalar_from_memref_subview
5460

5561
// -----
5662

57-
func.func @load_scalar_from_memref_static_dim_2(%input: memref<4x8xf32, strided<[8, 12], offset: 100>>, %row: index, %col: index, %value: f32) {
63+
func.func @store_scalar_from_memref_static_dim_2(%input: memref<4x8xf32, strided<[8, 12], offset: 100>>, %row: index, %col: index, %value: f32) {
5864
memref.store %value, %input[%row, %col] : memref<4x8xf32, strided<[8, 12], offset: 100>>
5965
return
6066
}
67+
// CHECK-LABEL: func @store_scalar_from_memref_static_dim_2
6168

6269
// -----
6370

6471
func.func @store_scalar_from_memref_dynamic_dim_2(%input: memref<4x8xf32, strided<[?, ?], offset: ?>>, %row: index, %col: index, %value: f32) {
6572
memref.store %value, %input[%row, %col] : memref<4x8xf32, strided<[?, ?], offset: ?>>
6673
return
6774
}
75+
// CHECK-LABEL: func @store_scalar_from_memref_dynamic_dim_2
6876

6977
// -----
7078

@@ -74,6 +82,7 @@ func.func @load_vector_from_memref(%input: memref<4x8xf32>) -> vector<8xf32> {
7482
%value = vector.load %input[%c3, %c6] : memref<4x8xf32>, vector<8xf32>
7583
return %value : vector<8xf32>
7684
}
85+
// CHECK-LABEL: func @load_vector_from_memref
7786

7887
// -----
7988

@@ -83,13 +92,15 @@ func.func @load_vector_from_memref_odd(%input: memref<3x7xi2>) -> vector<3xi2> {
8392
%value = vector.load %input[%c1, %c3] : memref<3x7xi2>, vector<3xi2>
8493
return %value : vector<3xi2>
8594
}
95+
// CHECK-LABEL: func @load_vector_from_memref_odd
8696

8797
// -----
8898

8999
func.func @load_vector_from_memref_dynamic(%input: memref<3x7xi2>, %row: index, %col: index) -> vector<3xi2> {
90100
%value = vector.load %input[%row, %col] : memref<3x7xi2>, vector<3xi2>
91101
return %value : vector<3xi2>
92102
}
103+
// CHECK-LABEL: func @load_vector_from_memref_dynamic
93104

94105
// -----
95106

@@ -99,10 +110,12 @@ func.func @store_vector_to_memref_odd(%input: memref<3x7xi2>, %value: vector<3xi
99110
vector.store %value, %input[%c1, %c3] : memref<3x7xi2>, vector<3xi2>
100111
return
101112
}
113+
// CHECK-LABEL: func @store_vector_to_memref_odd
102114

103115
// -----
104116

105117
func.func @store_vector_to_memref_dynamic(%input: memref<3x7xi2>, %value: vector<3xi2>, %row: index, %col: index) {
106118
vector.store %value, %input[%row, %col] : memref<3x7xi2>, vector<3xi2>
107119
return
108120
}
121+
// CHECK-LABEL: func @store_vector_to_memref_dynamic

0 commit comments

Comments
 (0)