1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
// RUN: mlir-opt %s -test-scf-while-op-builder | FileCheck %s
// CHECK-LABEL: @testMatchingTypes
func.func @testMatchingTypes(%arg0 : i32) {
%0 = scf.while (%arg1 = %arg0) : (i32) -> (i32) {
%c10 = arith.constant 10 : i32
%1 = arith.cmpi slt, %arg1, %c10 : i32
scf.condition(%1) %arg1 : i32
} do {
^bb0(%arg1: i32):
scf.yield %arg1 : i32
}
// Expect the same loop twice (the dummy added by the test pass and the
// original one).
// CHECK: %[[V0:.*]] = scf.while (%[[arg1:.*]] = %[[arg0:.*]]) : (i32) -> i32 {
// CHECK: %[[V1:.*]] = scf.while (%[[arg2:.*]] = %[[arg0]]) : (i32) -> i32 {
return
}
// CHECK-LABEL: @testNonMatchingTypes
func.func @testNonMatchingTypes(%arg0 : i32) {
%c1 = arith.constant 1 : i32
%c10 = arith.constant 10 : i32
%0:2 = scf.while (%arg1 = %arg0) : (i32) -> (i32, i32) {
%1 = arith.cmpi slt, %arg1, %c10 : i32
scf.condition(%1) %arg1, %c1 : i32, i32
} do {
^bb0(%arg1: i32, %arg2: i32):
%1 = arith.addi %arg1, %arg2 : i32
scf.yield %1 : i32
}
// Expect the same loop twice (the dummy added by the test pass and the
// original one).
// CHECK: %[[V0:.*]] = scf.while (%[[arg1:.*]] = %[[arg0:.*]]) : (i32) -> (i32, i32) {
// CHECK: %[[V1:.*]] = scf.while (%[[arg2:.*]] = %[[arg0]]) : (i32) -> (i32, i32) {
return
}
|
