Skip to content

Commit f2a5ed4

Browse files
committed
fixup! [mlir][vector] Refactor vector-transfer-flatten.mlir (nfc) (3/n)
Minor updates - addressing PR comments
1 parent 2b99b7c commit f2a5ed4

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

mlir/test/Dialect/Vector/vector-transfer-flatten.mlir

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -110,12 +110,12 @@ func.func @transfer_read_dims_mismatch_non_zero_indices(
110110

111111
func.func @transfer_read_dims_mismatch_non_contiguous_non_zero_indices(
112112
%arg : memref<1x3x3x2xf32, strided<[40, 10, 2, 1], offset: ?>>,
113-
%idx0 : index,
114-
%idx1 : index) -> vector<2x2xf32> {
113+
%idx_1 : index,
114+
%idx_2 : index) -> vector<2x2xf32> {
115115

116116
%c0 = arith.constant 0 : index
117117
%cst_1 = arith.constant 0.000000e+00 : f32
118-
%8 = vector.transfer_read %arg[%c0, %idx0, %idx1, %c0], %cst_1 {in_bounds = [true, true]} :
118+
%8 = vector.transfer_read %arg[%c0, %idx_1, %idx_2, %c0], %cst_1 {in_bounds = [true, true]} :
119119
memref<1x3x3x2xf32, strided<[40, 10, 2, 1], offset: ?>>, vector<2x2xf32>
120120
return %8 : vector<2x2xf32>
121121
}
@@ -358,11 +358,11 @@ func.func @transfer_write_dims_mismatch_non_zero_indices(
358358
func.func @transfer_write_dims_mismatch_non_contiguous_non_zero_indices(
359359
%value : vector<2x2xf32>,
360360
%subview : memref<1x3x3x2xf32, strided<[40, 10, 2, 1], offset: ?>>,
361-
%idx0 : index,
362-
%idx1 : index) {
361+
%idx_1 : index,
362+
%idx_2 : index) {
363363

364364
%c0 = arith.constant 0 : index
365-
vector.transfer_write %value, %subview[%c0, %idx0, %idx1, %c0] {in_bounds = [true, true]} : vector<2x2xf32>, memref<1x3x3x2xf32, strided<[40, 10, 2, 1], offset: ?>>
365+
vector.transfer_write %value, %subview[%c0, %idx_1, %idx_2, %c0] {in_bounds = [true, true]} : vector<2x2xf32>, memref<1x3x3x2xf32, strided<[40, 10, 2, 1], offset: ?>>
366366
return
367367
}
368368

@@ -392,7 +392,7 @@ func.func @transfer_write_leading_dynamic_dims(
392392
}
393393

394394
// CHECK-LABEL: func @transfer_write_leading_dynamic_dims
395-
// CHECK-SAME: %[[ARG0:.+]]: vector<8x4xi8>, %[[ARG1:.+]]: memref<?x?x8x4xi8, {{.+}}>, %[[ARG2:.+]]: index, %[[ARG3:.+]]: index
395+
// CHECK-SAME: %[[ARG0:.+]]: vector<8x4xi8>, %[[ARG1:.+]]: memref<?x?x8x4xi8, {{.+}}>, %[[ARG2:.+]]: index, %[[ARG3:.+]]: index
396396
// CHECK: %[[C0:.+]] = arith.constant 0 : index
397397
// CHECK: %[[COLLAPSED:.+]] = memref.collapse_shape %[[ARG1]] {{\[}}[0], [1], [2, 3]{{\]}}
398398
// CHECK-SAME: : memref<?x?x8x4xi8, {{.+}}> into memref<?x?x32xi8, {{.+}}>

0 commit comments

Comments
 (0)