45 lines
1.7 KiB
MLIR
45 lines
1.7 KiB
MLIR
// RUN: onnx-mlir-opt --lower-krnl %s -split-input-file | FileCheck %s
|
|
|
|
// CHECK-DAG: #{{.*}} = affine_map<(d0) -> (d0)>
|
|
// CHECK-DAG: #{{.*}} = affine_map<(d0) -> (d0 + 2)>
|
|
// CHECK-DAG: #{{.*}} = affine_map<() -> (0)>
|
|
// CHECK-DAG: #{{.*}} = affine_map<() -> (10)>
|
|
// CHECK-DAG: #{{.*}} = affine_map<(d0, d1) -> (d1 + 2, d0 + 4, 10)>
|
|
// CHECK-DAG: #{{.*}} = affine_map<(d0) -> (d0 + 4, 10)>
|
|
|
|
func @simple_block() {
|
|
// CHECK-LABEL: simple_block
|
|
// CHECK-NEXT: affine.for [[OUTER_LOOP:%.+]] = 0 to 10 step 2 {
|
|
// CHECK-NEXT: affine.for [[INNER_LOOP:%.+]] = #map{{.*}}([[OUTER_LOOP]]) to #map{{.*}}([[OUTER_LOOP]]) {
|
|
// CHECK-NEXT: %0 = addi [[INNER_LOOP]], [[INNER_LOOP]] : index
|
|
// CHECK-NEXT: }
|
|
// CHECK-NEXT: }
|
|
|
|
%ii = krnl.define_loops 1
|
|
%ib, %il = krnl.block %ii 2 : (!krnl.loop) -> (!krnl.loop, !krnl.loop)
|
|
krnl.iterate(%ib, %il) with (%ii -> %i = 0 to 10) {
|
|
%foo = addi %i, %i : index
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func @block_nested() {
|
|
// CHECK-LABEL: block_nested
|
|
// CHECK-NEXT: affine.for [[OUTER_LOOP:%.+]] = 0 to 10 step 4 {
|
|
// CHECK-NEXT: affine.for [[MIDDLE_LOOP:%.+]] = #map{{.*}}([[OUTER_LOOP]]) to min #map{{.*}}([[OUTER_LOOP]]) step 2 {
|
|
// CHECK-NEXT: affine.for [[INNER_LOOP:%.+]] = #map{{.*}}([[MIDDLE_LOOP]]) to min #map{{.*}}([[OUTER_LOOP]], [[MIDDLE_LOOP]]) {
|
|
// CHECK-NEXT: %0 = addi [[INNER_LOOP]], [[INNER_LOOP]] : index
|
|
// CHECK-NEXT: }
|
|
// CHECK-NEXT: }
|
|
// CHECK-NEXT: }
|
|
|
|
%ii = krnl.define_loops 1
|
|
%ib, %il = krnl.block %ii 4 : (!krnl.loop) -> (!krnl.loop, !krnl.loop)
|
|
%ilb, %ill = krnl.block %il 2 : (!krnl.loop) -> (!krnl.loop, !krnl.loop)
|
|
krnl.iterate(%ib, %ilb, %ill) with (%ii -> %i = 0 to 10) {
|
|
%foo = addi %i, %i : index
|
|
}
|
|
|
|
return
|
|
} |