diff options
Diffstat (limited to 'mlir/test/Transforms/loop-tiling.mlir')
| -rw-r--r-- | mlir/test/Transforms/loop-tiling.mlir | 36 |
1 files changed, 18 insertions, 18 deletions
diff --git a/mlir/test/Transforms/loop-tiling.mlir b/mlir/test/Transforms/loop-tiling.mlir index a1f9d717fab..c2fdbd4f80f 100644 --- a/mlir/test/Transforms/loop-tiling.mlir +++ b/mlir/test/Transforms/loop-tiling.mlir @@ -8,12 +8,12 @@ // CHECK-DAG: [[UB_INTRA_TILE:#map[0-9]+]] = (d0, d1, d2) -> (d2 + 32, s0, 4096 floordiv s1) // CHECK-LABEL: func @loop_tiling() -// CHECK-NEXT: affine.for %i0 = 0 to 256 step 32 { -// CHECK-NEXT: affine.for %i1 = 0 to 512 step 32 { -// CHECK-NEXT: affine.for %i2 = 0 to 1024 step 32 { -// CHECK-NEXT: affine.for %i3 = [[IDENTITY]](%i0) to [[MAP0]](%i0) { -// CHECK-NEXT: affine.for %i4 = [[IDENTITY]](%i1) to [[MAP0]](%i1) { -// CHECK-NEXT: affine.for %i5 = [[IDENTITY]](%i2) to [[MAP0]](%i2) { +// CHECK-NEXT: for %i0 = 0 to 256 step 32 { +// CHECK-NEXT: for %i1 = 0 to 512 step 32 { +// CHECK-NEXT: for %i2 = 0 to 1024 step 32 { +// CHECK-NEXT: for %i3 = [[IDENTITY]](%i0) to [[MAP0]](%i0) { +// CHECK-NEXT: for %i4 = [[IDENTITY]](%i1) to [[MAP0]](%i1) { +// CHECK-NEXT: for %i5 = [[IDENTITY]](%i2) to [[MAP0]](%i2) { // CHECK-NEXT: "foo"(%i3, %i4, %i5) : (index, index, index) -> () // CHECK-NEXT: } // CHECK-NEXT: } @@ -21,32 +21,32 @@ // CHECK-NEXT: } // CHECK-NEXT: } // CHECK-NEXT: } -// CHECK-NEXT: affine.for %i6 = 0 to 50 step 32 { -// CHECK-NEXT: affine.for %i7 = [[IDENTITY]](%i6) to min [[MAP1]](%i6) { +// CHECK-NEXT: for %i6 = 0 to 50 step 32 { +// CHECK-NEXT: for %i7 = [[IDENTITY]](%i6) to min [[MAP1]](%i6) { // CHECK-NEXT: "bar"(%i7, %i7) : (index, index) -> () // CHECK-NEXT: } // CHECK-NEXT: } -// CHECK-NEXT: affine.for %i8 = 0 to 21 step 32 { -// CHECK-NEXT: affine.for %i9 = [[IDENTITY]](%i8) to 21 { +// CHECK-NEXT: for %i8 = 0 to 21 step 32 { +// CHECK-NEXT: for %i9 = [[IDENTITY]](%i8) to 21 { // CHECK-NEXT: "foobar"(%i9) : (index) -> () // CHECK-NEXT: } // CHECK-NEXT: } // CHECK-NEXT: return func @loop_tiling() { - affine.for %i = 0 to 256 { - affine.for %j = 0 to 512 { - affine.for %k = 0 to 1024 { + for %i = 0 to 256 { + for %j = 0 to 512 { + for %k = 0 to 1024 { "foo"(%i, %j, %k) : (index, index, index) -> () } } } - affine.for %x = 0 to 50 { + for %x = 0 to 50 { "bar"(%x, %x) : (index, index) -> () } // Intra-tile loop won't need a min expression. - affine.for %y = 0 to 21 { + for %y = 0 to 21 { "foobar"(%y) : (index) -> () } @@ -58,12 +58,12 @@ func @loop_tiling() { // CHECK-LABEL: func @loop_max_min_bound(%arg0: memref<?xi32>, %arg1: index, %arg2: index) { func @loop_max_min_bound(%A : memref<? x i32>, %L : index, %U : index) { %M = dim %A, 0 : memref<? x i32> - affine.for %iTT = max #lb()[%L] to min #ub()[%M, %U] { + for %iTT = max #lb()[%L] to min #ub()[%M, %U] { %out = affine.apply (d0) -> (d0) (%iTT) } return -// CHECK: affine.for %i0 = max [[LB]]()[%arg1] to min [[UB]]()[%0, %arg2] step 32 { -// CHECK-NEXT: affine.for %i1 = [[IDENTITY]](%i0) to min [[UB_INTRA_TILE]](%0, %arg2, %i0) { +// CHECK: for %i0 = max [[LB]]()[%arg1] to min [[UB]]()[%0, %arg2] step 32 { +// CHECK-NEXT: for %i1 = [[IDENTITY]](%i0) to min [[UB_INTRA_TILE]](%0, %arg2, %i0) { // CHECK-NEXT: %1 = affine.apply [[IDENTITY]](%i1) // CHECK-NEXT: } // CHECK-NEXT: } |
