diff options
| -rw-r--r-- | mlir/test/Transforms/canonicalize.mlir | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/mlir/test/Transforms/canonicalize.mlir b/mlir/test/Transforms/canonicalize.mlir index a6c23262907..16ee00923d2 100644 --- a/mlir/test/Transforms/canonicalize.mlir +++ b/mlir/test/Transforms/canonicalize.mlir @@ -703,28 +703,28 @@ func @view(%arg0 : index) { %c15 = constant 15 : index // Test: fold constant sizes and offset, update map with static stride/offset. - // CHECK: std.view %0[][] : memref<2048xi8> to memref<7x11xf32, #[[VIEW_MAP0]]> + // CHECK: std.view %[[ALLOC_MEM]][][] : memref<2048xi8> to memref<7x11xf32, #[[VIEW_MAP0]]> %1 = view %0[%c15][%c7, %c11] : memref<2048xi8> to memref<?x?xf32, #TEST_VIEW_MAP0> load %1[%c0, %c0] : memref<?x?xf32, #TEST_VIEW_MAP0> // Test: fold constant sizes but not offset, update map with static stride. // Test that we do not a fold dynamic dim which is not produced by a constant. - // CHECK: std.view %0[%arg0][] : memref<2048xi8> to memref<7x11xf32, #[[VIEW_MAP1]]> + // CHECK: std.view %[[ALLOC_MEM]][%arg0][] : memref<2048xi8> to memref<7x11xf32, #[[VIEW_MAP1]]> %2 = view %0[%arg0][%c7, %c11] : memref<2048xi8> to memref<?x?xf32, #TEST_VIEW_MAP0> load %2[%c0, %c0] : memref<?x?xf32, #TEST_VIEW_MAP0> // Test: fold constant offset but not sizes, update map with constant offset. // Test that we fold constant offset but not dynamic dims. - // CHECK: std.view %0[][%arg0, %arg0] : memref<2048xi8> to memref<?x?xf32, #[[VIEW_MAP2]]> + // CHECK: std.view %[[ALLOC_MEM]][][%arg0, %arg0] : memref<2048xi8> to memref<?x?xf32, #[[VIEW_MAP2]]> %3 = view %0[%c15][%arg0, %arg0] : memref<2048xi8> to memref<?x?xf32, #TEST_VIEW_MAP0> load %3[%c0, %c0] : memref<?x?xf32, #TEST_VIEW_MAP0> // Test: fold one constant dim, no offset, should update with constant // stride on dim 1, but leave dynamic stride on dim 0. - // CHECK: std.view %0[][%arg0, %arg0] : memref<2048xi8> to memref<?x?x7xf32, #[[VIEW_MAP3]]> + // CHECK: std.view %[[ALLOC_MEM]][][%arg0, %arg0] : memref<2048xi8> to memref<?x?x7xf32, #[[VIEW_MAP3]]> %4 = view %0[][%arg0, %arg0, %c7] : memref<2048xi8> to memref<?x?x?xf32, #TEST_VIEW_MAP1> load %4[%c0, %c0, %c0] : memref<?x?x?xf32, #TEST_VIEW_MAP1> @@ -736,7 +736,7 @@ func @view(%arg0 : index) { load %5[%c0, %c0] : memref<?x4xf32, #TEST_VIEW_MAP2> // Test: folding static alloc and memref_cast into a view. - // CHECK: std.view %0[][%c15, %c7] : memref<2048xi8> to memref<?x?xf32> + // CHECK: std.view %[[ALLOC_MEM]][][%c15, %c7] : memref<2048xi8> to memref<?x?xf32> %6 = memref_cast %0 : memref<2048xi8> to memref<?xi8> %7 = view %6[%c15][%c7] : memref<?xi8> to memref<?x?xf32> load %7[%c0, %c0] : memref<?x?xf32> |
