Skip to content

Commit f60ba58

Browse files
authored
[NFC][GlobalOpt] Update function names in LIT (#22083)
Following up on commit dd684c4, rename the MLIR functions in the test, echoing the newly supported generalizations. Signed-off-by: Artem Gindinson <[email protected]>
1 parent c612a1f commit f60ba58

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

compiler/src/iree/compiler/GlobalOptimization/test/generalize_named_ops.mlir

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ util.func public @generalize_1x1_group_conv_2d(%input: tensor<1x2x3x4x5xf32>, %f
116116

117117
// -----
118118

119-
util.func public @no_generalize_1x1_conv_2d_strides(%input: tensor<1x7x7x2xf32>, %filter: tensor<1x1x2x7xf32>) -> tensor<1x4x4x7xf32> {
119+
util.func public @generalize_1x1_conv_2d_strides(%input: tensor<1x7x7x2xf32>, %filter: tensor<1x1x2x7xf32>) -> tensor<1x4x4x7xf32> {
120120
%0 = tensor.empty() : tensor<1x4x4x7xf32>
121121
%1 = linalg.conv_2d_nhwc_hwcf {
122122
dilations = dense<1> : tensor<2xi64>,
@@ -125,13 +125,13 @@ util.func public @no_generalize_1x1_conv_2d_strides(%input: tensor<1x7x7x2xf32>,
125125
util.return %1 : tensor<1x4x4x7xf32>
126126
}
127127

128-
// CHECK-LABEL: @no_generalize_1x1_conv_2d_strides
128+
// CHECK-LABEL: @generalize_1x1_conv_2d_strides
129129
// CHECK: linalg.generic
130130
// CHECK: util.return
131131

132132
// -----
133133

134-
util.func public @no_generalize_1x1_depthwise_conv(%input: tensor<1x2x3x4xf32>, %filter: tensor<1x1x4xf32>) -> tensor<1x2x3x4xf32> {
134+
util.func public @generalize_1x1_depthwise_conv(%input: tensor<1x2x3x4xf32>, %filter: tensor<1x1x4xf32>) -> tensor<1x2x3x4xf32> {
135135
%0 = tensor.empty() : tensor<1x2x3x4xf32>
136136
%1 = linalg.depthwise_conv_2d_nhwc_hwc {
137137
dilations = dense<1> : tensor<2xi64>,
@@ -140,6 +140,6 @@ util.func public @no_generalize_1x1_depthwise_conv(%input: tensor<1x2x3x4xf32>,
140140
util.return %1 : tensor<1x2x3x4xf32>
141141
}
142142

143-
// CHECK-LABEL: @no_generalize_1x1_depthwise_conv
143+
// CHECK-LABEL: @generalize_1x1_depthwise_conv
144144
// CHECK: linalg.generic
145145
// CHECK: util.return

0 commit comments

Comments
 (0)