diff options
Diffstat (limited to 'mlir/test/AffineOps/ops.mlir')
-rw-r--r-- | mlir/test/AffineOps/ops.mlir | 24 |
1 files changed, 12 insertions, 12 deletions
diff --git a/mlir/test/AffineOps/ops.mlir b/mlir/test/AffineOps/ops.mlir index d78ddd2d76f..48b21fa9e0e 100644 --- a/mlir/test/AffineOps/ops.mlir +++ b/mlir/test/AffineOps/ops.mlir @@ -22,7 +22,7 @@ func @empty() { // GENERIC-NEXT: "affine.terminator"() : () -> () // GENERIC-NEXT: }, { // GENERIC-NEXT: }) - affine.if () : () () { + affine.if affine_set<() : ()> () { } {some_attr = true} // CHECK: } else { @@ -34,7 +34,7 @@ func @empty() { // GENERIC-NEXT: "foo"() : () -> () // GENERIC-NEXT: "affine.terminator"() : () -> () // GENERIC-NEXT: }) - affine.if () : () () { + affine.if affine_set<() : ()> () { } else { "foo"() : () -> () } {some_attr = true} @@ -61,21 +61,21 @@ func @affine_terminator() { // ----- -// CHECK-DAG: #[[MAP0:map[0-9]+]] = (d0)[s0] -> (1000, d0 + 512, s0) -// CHECK-DAG: #[[MAP1:map[0-9]+]] = (d0, d1)[s0] -> (d0 - d1, s0 + 512) -// CHECK-DAG: #[[MAP2:map[0-9]+]] = ()[s0, s1] -> (s0 - s1, 11) -// CHECK-DAG: #[[MAP3:map[0-9]+]] = () -> (77, 78, 79) +// CHECK-DAG: #[[MAP0:map[0-9]+]] = affine_map<(d0)[s0] -> (1000, d0 + 512, s0)> +// CHECK-DAG: #[[MAP1:map[0-9]+]] = affine_map<(d0, d1)[s0] -> (d0 - d1, s0 + 512)> +// CHECK-DAG: #[[MAP2:map[0-9]+]] = affine_map<()[s0, s1] -> (s0 - s1, 11)> +// CHECK-DAG: #[[MAP3:map[0-9]+]] = affine_map<() -> (77, 78, 79)> // CHECK-LABEL: @affine_min func @affine_min(%arg0 : index, %arg1 : index, %arg2 : index) { // CHECK: affine.min #[[MAP0]](%arg0)[%arg1] - %0 = affine.min (d0)[s0] -> (1000, d0 + 512, s0) (%arg0)[%arg1] + %0 = affine.min affine_map<(d0)[s0] -> (1000, d0 + 512, s0)> (%arg0)[%arg1] // CHECK: affine.min #[[MAP1]](%arg0, %arg1)[%arg2] - %1 = affine.min (d0, d1)[s0] -> (d0 - d1, s0 + 512) (%arg0, %arg1)[%arg2] + %1 = affine.min affine_map<(d0, d1)[s0] -> (d0 - d1, s0 + 512)> (%arg0, %arg1)[%arg2] // CHECK: affine.min #[[MAP2]]()[%arg1, %arg2] - %2 = affine.min ()[s0, s1] -> (s0 - s1, 11) ()[%arg1, %arg2] + %2 = affine.min affine_map<()[s0, s1] -> (s0 - s1, 11)> ()[%arg1, %arg2] // CHECK: affine.min #[[MAP3]]() - %3 = affine.min ()[] -> (77, 78, 79) ()[] + %3 = affine.min affine_map<()[] -> (77, 78, 79)> ()[] return } @@ -90,8 +90,8 @@ func @valid_symbols(%arg0: index, %arg1: index, %arg2: index) { affine.for %arg4 = 0 to %13 step 264 { %18 = dim %0, 0 : memref<?x?xf32> %20 = std.subview %0[%c0, %c0][%18,%arg4][%c1,%c1] : memref<?x?xf32> - to memref<?x?xf32, (d0, d1)[s0, s1, s2] -> (d0 * s1 + d1 * s2 + s0)> - %24 = dim %20, 0 : memref<?x?xf32, (d0, d1)[s0, s1, s2] -> (d0 * s1 + d1 * s2 + s0)> + to memref<?x?xf32, affine_map<(d0, d1)[s0, s1, s2] -> (d0 * s1 + d1 * s2 + s0)>> + %24 = dim %20, 0 : memref<?x?xf32, affine_map<(d0, d1)[s0, s1, s2] -> (d0 * s1 + d1 * s2 + s0)>> affine.for %arg5 = 0 to %24 step 768 { "foo"() : () -> () } |