summaryrefslogtreecommitdiff
path: root/mlir/test/Target/Wasm/double_nested_loop.mlir
blob: 8b3e4997e63e18cb030d8d4603369d57af5e0cd4 (plain)
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
// RUN: yaml2obj %S/inputs/double_nested_loop.yaml.wasm -o - | mlir-translate --import-wasm | FileCheck %s

/*
(module
  (func
    ;; create a local variable and initialize it to 0
    (local $i i32)
    (local $j i32)

    (loop $my_loop

      ;; add one to $i
      local.get $i
      i32.const 1
      i32.add
      local.set $i
      (loop $my_second_loop (result i32)
        i32.const 1
        local.get $j
        i32.const 12
        i32.add
        local.tee $j
        local.get $i
        i32.gt_s
        br_if $my_second_loop
      )
      i32.const 10
      i32.lt_s
      br_if $my_loop
    )
  )
)
*/

// CHECK-LABEL:   wasmssa.func @func_0() {
// CHECK:           %[[VAL_0:.*]] = wasmssa.local of type i32
// CHECK:           %[[VAL_1:.*]] = wasmssa.local of type i32
// CHECK:           wasmssa.loop : {
// CHECK:             %[[VAL_2:.*]] = wasmssa.local_get %[[VAL_0]] :  ref to i32
// CHECK:             %[[VAL_3:.*]] = wasmssa.const 1 : i32
// CHECK:             %[[VAL_4:.*]] = wasmssa.add %[[VAL_2]] %[[VAL_3]] : i32
// CHECK:             wasmssa.local_set %[[VAL_0]] :  ref to i32 to %[[VAL_4]] : i32
// CHECK:             wasmssa.loop : {
// CHECK:               %[[VAL_5:.*]] = wasmssa.const 1 : i32
// CHECK:               %[[VAL_6:.*]] = wasmssa.local_get %[[VAL_1]] :  ref to i32
// CHECK:               %[[VAL_7:.*]] = wasmssa.const 12 : i32
// CHECK:               %[[VAL_8:.*]] = wasmssa.add %[[VAL_6]] %[[VAL_7]] : i32
// CHECK:               %[[VAL_9:.*]] = wasmssa.local_tee %[[VAL_1]] :  ref to i32 to %[[VAL_8]] : i32
// CHECK:               %[[VAL_10:.*]] = wasmssa.local_get %[[VAL_0]] :  ref to i32
// CHECK:               %[[VAL_11:.*]] = wasmssa.gt_si %[[VAL_9]] %[[VAL_10]] : i32 -> i32
// CHECK:               wasmssa.branch_if %[[VAL_11]] to level 0 else ^bb1
// CHECK:             ^bb1:
// CHECK:               wasmssa.block_return %[[VAL_5]] : i32
// CHECK:             }> ^bb1
// CHECK:           ^bb1(%[[VAL_12:.*]]: i32):
// CHECK:             %[[VAL_13:.*]] = wasmssa.const 10 : i32
// CHECK:             %[[VAL_14:.*]] = wasmssa.lt_si %[[VAL_12]] %[[VAL_13]] : i32 -> i32
// CHECK:             wasmssa.branch_if %[[VAL_14]] to level 0 else ^bb2
// CHECK:           ^bb2:
// CHECK:             wasmssa.block_return
// CHECK:           }> ^bb1
// CHECK:         ^bb1:
// CHECK:           wasmssa.return