test/lit/passes/remove-unused-brs_levels.wast

;; NOTE: Assertions have been generated by update_lit_checks.py and should not be edited.
;; RUN: wasm-opt %s --remove-unused-brs -all -S --shrink-level=0 -o - | filecheck %s --check-prefix=SHRINK_0
;; RUN: wasm-opt %s --remove-unused-brs -all -S --shrink-level=1 -o - | filecheck %s --check-prefix=SHRINK_1
;; RUN: wasm-opt %s --remove-unused-brs -all -S --shrink-level=2 -o - | filecheck %s --check-prefix=SHRINK_2

(module
  ;; SHRINK_0:      (type $struct (sub (struct)))
  ;; SHRINK_1:      (type $struct (sub (struct)))
  ;; SHRINK_2:      (type $struct (sub (struct)))
  (type $struct (sub (struct)))
  ;; SHRINK_0:      (type $substruct (sub $struct (struct)))
  ;; SHRINK_1:      (type $substruct (sub $struct (struct)))
  ;; SHRINK_2:      (type $substruct (sub $struct (struct)))
  (type $substruct (sub $struct (struct)))

  ;; SHRINK_0:      (func $selectify-division (type $2) (param $x i32) (result i32)
  ;; SHRINK_0-NEXT:  (if (result i32)
  ;; SHRINK_0-NEXT:   (i32.eq
  ;; SHRINK_0-NEXT:    (local.get $x)
  ;; SHRINK_0-NEXT:    (i32.const 53498923)
  ;; SHRINK_0-NEXT:   )
  ;; SHRINK_0-NEXT:   (then
  ;; SHRINK_0-NEXT:    (i32.div_s
  ;; SHRINK_0-NEXT:     (local.get $x)
  ;; SHRINK_0-NEXT:     (i32.const 13)
  ;; SHRINK_0-NEXT:    )
  ;; SHRINK_0-NEXT:   )
  ;; SHRINK_0-NEXT:   (else
  ;; SHRINK_0-NEXT:    (local.get $x)
  ;; SHRINK_0-NEXT:   )
  ;; SHRINK_0-NEXT:  )
  ;; SHRINK_0-NEXT: )
  ;; SHRINK_1:      (func $selectify-division (type $2) (param $x i32) (result i32)
  ;; SHRINK_1-NEXT:  (select
  ;; SHRINK_1-NEXT:   (i32.div_s
  ;; SHRINK_1-NEXT:    (local.get $x)
  ;; SHRINK_1-NEXT:    (i32.const 13)
  ;; SHRINK_1-NEXT:   )
  ;; SHRINK_1-NEXT:   (local.get $x)
  ;; SHRINK_1-NEXT:   (i32.eq
  ;; SHRINK_1-NEXT:    (local.get $x)
  ;; SHRINK_1-NEXT:    (i32.const 53498923)
  ;; SHRINK_1-NEXT:   )
  ;; SHRINK_1-NEXT:  )
  ;; SHRINK_1-NEXT: )
  ;; SHRINK_2:      (func $selectify-division (type $2) (param $x i32) (result i32)
  ;; SHRINK_2-NEXT:  (select
  ;; SHRINK_2-NEXT:   (i32.div_s
  ;; SHRINK_2-NEXT:    (local.get $x)
  ;; SHRINK_2-NEXT:    (i32.const 13)
  ;; SHRINK_2-NEXT:   )
  ;; SHRINK_2-NEXT:   (local.get $x)
  ;; SHRINK_2-NEXT:   (i32.eq
  ;; SHRINK_2-NEXT:    (local.get $x)
  ;; SHRINK_2-NEXT:    (i32.const 53498923)
  ;; SHRINK_2-NEXT:   )
  ;; SHRINK_2-NEXT:  )
  ;; SHRINK_2-NEXT: )
  (func $selectify-division (param $x i32) (result i32)
    ;; See #5983: this if, if turned into a select, becomes almost 5x slower.
    ;; We only want to selectify here when the shrink level is 1 or 2.
    (if (result i32)
      (i32.eq
        (local.get $x)
        (i32.const 53498923)
      )
      (then
        (i32.div_s
          (local.get $x)
          (i32.const 13)
        )
      )
      (else
        (local.get $x)
      )
    )
  )

  ;; SHRINK_0:      (func $selectify-division2 (type $2) (param $x i32) (result i32)
  ;; SHRINK_0-NEXT:  (if (result i32)
  ;; SHRINK_0-NEXT:   (i32.eq
  ;; SHRINK_0-NEXT:    (local.get $x)
  ;; SHRINK_0-NEXT:    (i32.const 53498923)
  ;; SHRINK_0-NEXT:   )
  ;; SHRINK_0-NEXT:   (then
  ;; SHRINK_0-NEXT:    (i32.div_s
  ;; SHRINK_0-NEXT:     (i32.div_s
  ;; SHRINK_0-NEXT:      (local.get $x)
  ;; SHRINK_0-NEXT:      (i32.const 13)
  ;; SHRINK_0-NEXT:     )
  ;; SHRINK_0-NEXT:     (i32.const 13)
  ;; SHRINK_0-NEXT:    )
  ;; SHRINK_0-NEXT:   )
  ;; SHRINK_0-NEXT:   (else
  ;; SHRINK_0-NEXT:    (local.get $x)
  ;; SHRINK_0-NEXT:   )
  ;; SHRINK_0-NEXT:  )
  ;; SHRINK_0-NEXT: )
  ;; SHRINK_1:      (func $selectify-division2 (type $2) (param $x i32) (result i32)
  ;; SHRINK_1-NEXT:  (if (result i32)
  ;; SHRINK_1-NEXT:   (i32.eq
  ;; SHRINK_1-NEXT:    (local.get $x)
  ;; SHRINK_1-NEXT:    (i32.const 53498923)
  ;; SHRINK_1-NEXT:   )
  ;; SHRINK_1-NEXT:   (then
  ;; SHRINK_1-NEXT:    (i32.div_s
  ;; SHRINK_1-NEXT:     (i32.div_s
  ;; SHRINK_1-NEXT:      (local.get $x)
  ;; SHRINK_1-NEXT:      (i32.const 13)
  ;; SHRINK_1-NEXT:     )
  ;; SHRINK_1-NEXT:     (i32.const 13)
  ;; SHRINK_1-NEXT:    )
  ;; SHRINK_1-NEXT:   )
  ;; SHRINK_1-NEXT:   (else
  ;; SHRINK_1-NEXT:    (local.get $x)
  ;; SHRINK_1-NEXT:   )
  ;; SHRINK_1-NEXT:  )
  ;; SHRINK_1-NEXT: )
  ;; SHRINK_2:      (func $selectify-division2 (type $2) (param $x i32) (result i32)
  ;; SHRINK_2-NEXT:  (select
  ;; SHRINK_2-NEXT:   (i32.div_s
  ;; SHRINK_2-NEXT:    (i32.div_s
  ;; SHRINK_2-NEXT:     (local.get $x)
  ;; SHRINK_2-NEXT:     (i32.const 13)
  ;; SHRINK_2-NEXT:    )
  ;; SHRINK_2-NEXT:    (i32.const 13)
  ;; SHRINK_2-NEXT:   )
  ;; SHRINK_2-NEXT:   (local.get $x)
  ;; SHRINK_2-NEXT:   (i32.eq
  ;; SHRINK_2-NEXT:    (local.get $x)
  ;; SHRINK_2-NEXT:    (i32.const 53498923)
  ;; SHRINK_2-NEXT:   )
  ;; SHRINK_2-NEXT:  )
  ;; SHRINK_2-NEXT: )
  (func $selectify-division2 (param $x i32) (result i32)
    ;; As above, but now only with a shrink level of 2 should we selectify, as
    ;; there are two divisions.
    (if (result i32)
      (i32.eq
        (local.get $x)
        (i32.const 53498923)
      )
      (then
        (i32.div_s
          (i32.div_s
            (local.get $x)
            (i32.const 13)
          )
          (i32.const 13)
        )
      )
      (else
        (local.get $x)
      )
    )
  )

  ;; SHRINK_0:      (func $if-tests (type $3) (param $x (ref $struct)) (param $y (ref $struct)) (result i32)
  ;; SHRINK_0-NEXT:  (if (result i32)
  ;; SHRINK_0-NEXT:   (ref.test (ref $substruct)
  ;; SHRINK_0-NEXT:    (local.get $x)
  ;; SHRINK_0-NEXT:   )
  ;; SHRINK_0-NEXT:   (then
  ;; SHRINK_0-NEXT:    (ref.test (ref $substruct)
  ;; SHRINK_0-NEXT:     (local.get $y)
  ;; SHRINK_0-NEXT:    )
  ;; SHRINK_0-NEXT:   )
  ;; SHRINK_0-NEXT:   (else
  ;; SHRINK_0-NEXT:    (i32.const 0)
  ;; SHRINK_0-NEXT:   )
  ;; SHRINK_0-NEXT:  )
  ;; SHRINK_0-NEXT: )
  ;; SHRINK_1:      (func $if-tests (type $3) (param $x (ref $struct)) (param $y (ref $struct)) (result i32)
  ;; SHRINK_1-NEXT:  (select
  ;; SHRINK_1-NEXT:   (ref.test (ref $substruct)
  ;; SHRINK_1-NEXT:    (local.get $y)
  ;; SHRINK_1-NEXT:   )
  ;; SHRINK_1-NEXT:   (i32.const 0)
  ;; SHRINK_1-NEXT:   (ref.test (ref $substruct)
  ;; SHRINK_1-NEXT:    (local.get $x)
  ;; SHRINK_1-NEXT:   )
  ;; SHRINK_1-NEXT:  )
  ;; SHRINK_1-NEXT: )
  ;; SHRINK_2:      (func $if-tests (type $3) (param $x (ref $struct)) (param $y (ref $struct)) (result i32)
  ;; SHRINK_2-NEXT:  (select
  ;; SHRINK_2-NEXT:   (ref.test (ref $substruct)
  ;; SHRINK_2-NEXT:    (local.get $y)
  ;; SHRINK_2-NEXT:   )
  ;; SHRINK_2-NEXT:   (i32.const 0)
  ;; SHRINK_2-NEXT:   (ref.test (ref $substruct)
  ;; SHRINK_2-NEXT:    (local.get $x)
  ;; SHRINK_2-NEXT:   )
  ;; SHRINK_2-NEXT:  )
  ;; SHRINK_2-NEXT: )
  (func $if-tests (param $x (ref $struct)) (param $y (ref $struct)) (result i32)
    ;; An If with a test in one arm (and also in a condition). We do not
    ;; normally turn this into a Select, as the test is more costly than the If.
    ;; But we do so when optimizing for size.
    ;;
    ;; TODO: Consider optimizing this because the other arm is a 0, which means
    ;;       the Select can turn into an And later.
    (if (result i32)
      (ref.test (ref $substruct)
        (local.get $x)
      )
      (then
        (ref.test (ref $substruct)
          (local.get $y)
        )
      )
      (else
        (i32.const 0)
      )
    )
  )

  ;; SHRINK_0:      (func $if-tests-arms (type $4) (param $x (ref $struct)) (param $y (ref $struct)) (param $z (ref $struct)) (result i32)
  ;; SHRINK_0-NEXT:  (if (result i32)
  ;; SHRINK_0-NEXT:   (ref.test (ref $substruct)
  ;; SHRINK_0-NEXT:    (local.get $x)
  ;; SHRINK_0-NEXT:   )
  ;; SHRINK_0-NEXT:   (then
  ;; SHRINK_0-NEXT:    (ref.test (ref $substruct)
  ;; SHRINK_0-NEXT:     (local.get $y)
  ;; SHRINK_0-NEXT:    )
  ;; SHRINK_0-NEXT:   )
  ;; SHRINK_0-NEXT:   (else
  ;; SHRINK_0-NEXT:    (ref.test (ref $substruct)
  ;; SHRINK_0-NEXT:     (local.get $z)
  ;; SHRINK_0-NEXT:    )
  ;; SHRINK_0-NEXT:   )
  ;; SHRINK_0-NEXT:  )
  ;; SHRINK_0-NEXT: )
  ;; SHRINK_1:      (func $if-tests-arms (type $4) (param $x (ref $struct)) (param $y (ref $struct)) (param $z (ref $struct)) (result i32)
  ;; SHRINK_1-NEXT:  (select
  ;; SHRINK_1-NEXT:   (ref.test (ref $substruct)
  ;; SHRINK_1-NEXT:    (local.get $y)
  ;; SHRINK_1-NEXT:   )
  ;; SHRINK_1-NEXT:   (ref.test (ref $substruct)
  ;; SHRINK_1-NEXT:    (local.get $z)
  ;; SHRINK_1-NEXT:   )
  ;; SHRINK_1-NEXT:   (ref.test (ref $substruct)
  ;; SHRINK_1-NEXT:    (local.get $x)
  ;; SHRINK_1-NEXT:   )
  ;; SHRINK_1-NEXT:  )
  ;; SHRINK_1-NEXT: )
  ;; SHRINK_2:      (func $if-tests-arms (type $4) (param $x (ref $struct)) (param $y (ref $struct)) (param $z (ref $struct)) (result i32)
  ;; SHRINK_2-NEXT:  (select
  ;; SHRINK_2-NEXT:   (ref.test (ref $substruct)
  ;; SHRINK_2-NEXT:    (local.get $y)
  ;; SHRINK_2-NEXT:   )
  ;; SHRINK_2-NEXT:   (ref.test (ref $substruct)
  ;; SHRINK_2-NEXT:    (local.get $z)
  ;; SHRINK_2-NEXT:   )
  ;; SHRINK_2-NEXT:   (ref.test (ref $substruct)
  ;; SHRINK_2-NEXT:    (local.get $x)
  ;; SHRINK_2-NEXT:   )
  ;; SHRINK_2-NEXT:  )
  ;; SHRINK_2-NEXT: )
  (func $if-tests-arms (param $x (ref $struct)) (param $y (ref $struct)) (param $z (ref $struct)) (result i32)
    ;; As above but now with a test in both arms. The outcomes are the same.
    (if (result i32)
      (ref.test (ref $substruct)
        (local.get $x)
      )
      (then
        (ref.test (ref $substruct)
          (local.get $y)
        )
      )
      (else
        (ref.test (ref $substruct)
          (local.get $z)
        )
      )
    )
  )
)