# NOTE: Assertions have been autogenerated by utils/update_mir_test_checks.py
# RUN: llc -o - -mtriple=aarch64-unknown-unknown -run-pass=aarch64-prelegalizer-combiner -verify-machineinstrs  %s | FileCheck %s

# Truncs with a single use get folded.

# and(trunc(x), trunc(y)) -> trunc(and(x, y))
---
name:            and_trunc
body:             |
  bb.0:
    liveins: $w0, $w1
    ; CHECK-LABEL: name: and_trunc
    ; CHECK: liveins: $w0, $w1
    ; CHECK-NEXT: {{  $}}
    ; CHECK-NEXT: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
    ; CHECK-NEXT: [[COPY1:%[0-9]+]]:_(s32) = COPY $w1
    ; CHECK-NEXT: [[AND:%[0-9]+]]:_(s32) = G_AND [[COPY]], [[COPY1]]
    ; CHECK-NEXT: $w0 = COPY [[AND]](s32)
    %0:_(s32) = COPY $w0
    %1:_(s32) = COPY $w1
    %2:_(s16) = G_TRUNC %0
    %3:_(s16) = G_TRUNC %1
    %4:_(s16) = G_AND %2, %3
    %5:_(s32) = G_ANYEXT %4
    $w0 = COPY %5
...
---
name:            and_trunc_vector
body:             |
  bb.0:
    liveins: $q0, $q1
    ; CHECK-LABEL: name: and_trunc_vector
    ; CHECK: liveins: $q0, $q1
    ; CHECK-NEXT: {{  $}}
    ; CHECK-NEXT: [[COPY:%[0-9]+]]:_(<4 x s32>) = COPY $q0
    ; CHECK-NEXT: [[COPY1:%[0-9]+]]:_(<4 x s32>) = COPY $q1
    ; CHECK-NEXT: [[AND:%[0-9]+]]:_(<4 x s32>) = G_AND [[COPY]], [[COPY1]]
    ; CHECK-NEXT: [[TRUNC:%[0-9]+]]:_(<4 x s16>) = G_TRUNC [[AND]](<4 x s32>)
    ; CHECK-NEXT: $x0 = COPY [[TRUNC]](<4 x s16>)
    %0:_(<4 x s32>) = COPY $q0
    %1:_(<4 x s32>) = COPY $q1
    %2:_(<4 x s16>) = G_TRUNC %0
    %3:_(<4 x s16>) = G_TRUNC %1
    %4:_(<4 x s16>) = G_AND %2, %3
    $x0 = COPY %4
...

# or(trunc(x), trunc(y)) -> trunc(or(x, y))
---
name:            or_trunc
body:             |
  bb.0:
    liveins: $w0, $w1
    ; CHECK-LABEL: name: or_trunc
    ; CHECK: liveins: $w0, $w1
    ; CHECK-NEXT: {{  $}}
    ; CHECK-NEXT: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
    ; CHECK-NEXT: [[COPY1:%[0-9]+]]:_(s32) = COPY $w1
    ; CHECK-NEXT: [[OR:%[0-9]+]]:_(s32) = G_OR [[COPY]], [[COPY1]]
    ; CHECK-NEXT: $w0 = COPY [[OR]](s32)
    %0:_(s32) = COPY $w0
    %1:_(s32) = COPY $w1
    %2:_(s16) = G_TRUNC %0
    %3:_(s16) = G_TRUNC %1
    %4:_(s16) = G_OR %2, %3
    %5:_(s32) = G_ANYEXT %4
    $w0 = COPY %5
...
---
name:            or_trunc_vector
body:             |
  bb.0:
    liveins: $q0, $q1
    ; CHECK-LABEL: name: or_trunc_vector
    ; CHECK: liveins: $q0, $q1
    ; CHECK-NEXT: {{  $}}
    ; CHECK-NEXT: [[COPY:%[0-9]+]]:_(<4 x s32>) = COPY $q0
    ; CHECK-NEXT: [[COPY1:%[0-9]+]]:_(<4 x s32>) = COPY $q1
    ; CHECK-NEXT: [[OR:%[0-9]+]]:_(<4 x s32>) = G_OR [[COPY]], [[COPY1]]
    ; CHECK-NEXT: [[TRUNC:%[0-9]+]]:_(<4 x s16>) = G_TRUNC [[OR]](<4 x s32>)
    ; CHECK-NEXT: $x0 = COPY [[TRUNC]](<4 x s16>)
    %0:_(<4 x s32>) = COPY $q0
    %1:_(<4 x s32>) = COPY $q1
    %2:_(<4 x s16>) = G_TRUNC %0
    %3:_(<4 x s16>) = G_TRUNC %1
    %4:_(<4 x s16>) = G_OR %2, %3
    $x0 = COPY %4
...

# xor(trunc(x), trunc(y)) -> trunc(xor(x, y))
---
name:            xor_trunc
body:             |
  bb.0:
    liveins: $w0, $w1
    ; CHECK-LABEL: name: xor_trunc
    ; CHECK: liveins: $w0, $w1
    ; CHECK-NEXT: {{  $}}
    ; CHECK-NEXT: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
    ; CHECK-NEXT: [[COPY1:%[0-9]+]]:_(s32) = COPY $w1
    ; CHECK-NEXT: [[XOR:%[0-9]+]]:_(s32) = G_XOR [[COPY]], [[COPY1]]
    ; CHECK-NEXT: $w0 = COPY [[XOR]](s32)
    %0:_(s32) = COPY $w0
    %1:_(s32) = COPY $w1
    %2:_(s16) = G_TRUNC %0
    %3:_(s16) = G_TRUNC %1
    %4:_(s16) = G_XOR %2, %3
    %5:_(s32) = G_ANYEXT %4
    $w0 = COPY %5
...
---
name:            xor_trunc_vector
body:             |
  bb.0:
    liveins: $q0, $q1
    ; CHECK-LABEL: name: xor_trunc_vector
    ; CHECK: liveins: $q0, $q1
    ; CHECK-NEXT: {{  $}}
    ; CHECK-NEXT: [[COPY:%[0-9]+]]:_(<4 x s32>) = COPY $q0
    ; CHECK-NEXT: [[COPY1:%[0-9]+]]:_(<4 x s32>) = COPY $q1
    ; CHECK-NEXT: [[XOR:%[0-9]+]]:_(<4 x s32>) = G_XOR [[COPY]], [[COPY1]]
    ; CHECK-NEXT: [[TRUNC:%[0-9]+]]:_(<4 x s16>) = G_TRUNC [[XOR]](<4 x s32>)
    ; CHECK-NEXT: $x0 = COPY [[TRUNC]](<4 x s16>)
    %0:_(<4 x s32>) = COPY $q0
    %1:_(<4 x s32>) = COPY $q1
    %2:_(<4 x s16>) = G_TRUNC %0
    %3:_(<4 x s16>) = G_TRUNC %1
    %4:_(<4 x s16>) = G_XOR %2, %3
    $x0 = COPY %4
...

# Truncs with multiple uses do not get folded.
---
name:            or_trunc_multiuse_1
body:             |
  bb.0:
    liveins: $w0, $w1, $x2
    ; CHECK-LABEL: name: or_trunc_multiuse_1
    ; CHECK: liveins: $w0, $w1, $x2
    ; CHECK-NEXT: {{  $}}
    ; CHECK-NEXT: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
    ; CHECK-NEXT: [[COPY1:%[0-9]+]]:_(s32) = COPY $w1
    ; CHECK-NEXT: [[COPY2:%[0-9]+]]:_(p0) = COPY $x2
    ; CHECK-NEXT: [[TRUNC:%[0-9]+]]:_(s16) = G_TRUNC [[COPY]](s32)
    ; CHECK-NEXT: [[TRUNC1:%[0-9]+]]:_(s16) = G_TRUNC [[COPY1]](s32)
    ; CHECK-NEXT: G_STORE [[TRUNC]](s16), [[COPY2]](p0) :: (store (s16))
    ; CHECK-NEXT: [[OR:%[0-9]+]]:_(s16) = G_OR [[TRUNC]], [[TRUNC1]]
    ; CHECK-NEXT: [[ANYEXT:%[0-9]+]]:_(s32) = G_ANYEXT [[OR]](s16)
    ; CHECK-NEXT: $w0 = COPY [[ANYEXT]](s32)
    %0:_(s32) = COPY $w0
    %1:_(s32) = COPY $w1
    %5:_(p0) = COPY $x2
    %2:_(s16) = G_TRUNC %0
    %3:_(s16) = G_TRUNC %1
    G_STORE %2, %5 :: (store (s16))
    %4:_(s16) = G_OR %2, %3
    %6:_(s32) = G_ANYEXT %4
    $w0 = COPY %6
...
---
name:            and_trunc_multiuse_2
body:             |
  bb.0:
    liveins: $w0, $w1, $x2
    ; CHECK-LABEL: name: and_trunc_multiuse_2
    ; CHECK: liveins: $w0, $w1, $x2
    ; CHECK-NEXT: {{  $}}
    ; CHECK-NEXT: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
    ; CHECK-NEXT: [[COPY1:%[0-9]+]]:_(s32) = COPY $w1
    ; CHECK-NEXT: [[COPY2:%[0-9]+]]:_(p0) = COPY $x2
    ; CHECK-NEXT: [[TRUNC:%[0-9]+]]:_(s16) = G_TRUNC [[COPY]](s32)
    ; CHECK-NEXT: [[TRUNC1:%[0-9]+]]:_(s16) = G_TRUNC [[COPY1]](s32)
    ; CHECK-NEXT: G_STORE [[TRUNC]](s16), [[COPY2]](p0) :: (store (s16))
    ; CHECK-NEXT: [[AND:%[0-9]+]]:_(s16) = G_AND [[TRUNC]], [[TRUNC1]]
    ; CHECK-NEXT: [[ANYEXT:%[0-9]+]]:_(s32) = G_ANYEXT [[AND]](s16)
    ; CHECK-NEXT: $w0 = COPY [[ANYEXT]](s32)
    %0:_(s32) = COPY $w0
    %1:_(s32) = COPY $w1
    %5:_(p0) = COPY $x2
    %2:_(s16) = G_TRUNC %0
    %3:_(s16) = G_TRUNC %1
    G_STORE %2, %5 :: (store (s16))
    %4:_(s16) = G_AND %2, %3
    %6:_(s32) = G_ANYEXT %4
    $w0 = COPY %6
...
---
name:            xor_trunc_vector_multiuse
body:             |
  bb.0:
    liveins: $w0, $w1, $x2
    ; CHECK-LABEL: name: xor_trunc_vector_multiuse
    ; CHECK: liveins: $w0, $w1, $x2
    ; CHECK-NEXT: {{  $}}
    ; CHECK-NEXT: [[COPY:%[0-9]+]]:_(<4 x s32>) = COPY $q0
    ; CHECK-NEXT: [[COPY1:%[0-9]+]]:_(<4 x s32>) = COPY $q1
    ; CHECK-NEXT: [[COPY2:%[0-9]+]]:_(p0) = COPY $x2
    ; CHECK-NEXT: [[TRUNC:%[0-9]+]]:_(<4 x s16>) = G_TRUNC [[COPY]](<4 x s32>)
    ; CHECK-NEXT: [[TRUNC1:%[0-9]+]]:_(<4 x s16>) = G_TRUNC [[COPY1]](<4 x s32>)
    ; CHECK-NEXT: G_STORE [[TRUNC]](<4 x s16>), [[COPY2]](p0) :: (store (<4 x s16>))
    ; CHECK-NEXT: [[XOR:%[0-9]+]]:_(<4 x s16>) = G_XOR [[TRUNC]], [[TRUNC1]]
    ; CHECK-NEXT: $x0 = COPY [[XOR]](<4 x s16>)
    %0:_(<4 x s32>) = COPY $q0
    %1:_(<4 x s32>) = COPY $q1
    %5:_(p0) = COPY $x2
    %2:_(<4 x s16>) = G_TRUNC %0
    %3:_(<4 x s16>) = G_TRUNC %1
    G_STORE %2, %5 :: (store (<4 x s16>))
    %4:_(<4 x s16>) = G_XOR %2, %3
    $x0 = COPY %4
...

# Freezes should get pushed through truncs.

# This optimizes the pattern where `select(cond, T, 0)` gets converted to
# `and(cond, freeze(T))`.

# and(freeze(trunc(x)), trunc(y)) -> trunc(and(freeze(x), y))
---
name:            and_trunc_freeze
body:             |
  bb.0:
    liveins: $w0, $w1
    ; CHECK-LABEL: name: and_trunc_freeze
    ; CHECK: liveins: $w0, $w1
    ; CHECK-NEXT: {{  $}}
    ; CHECK-NEXT: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
    ; CHECK-NEXT: [[COPY1:%[0-9]+]]:_(s32) = COPY $w1
    ; CHECK-NEXT: [[FREEZE:%[0-9]+]]:_(s32) = G_FREEZE [[COPY]]
    ; CHECK-NEXT: [[AND:%[0-9]+]]:_(s32) = G_AND [[FREEZE]], [[COPY1]]
    ; CHECK-NEXT: $w0 = COPY [[AND]](s32)
    %0:_(s32) = COPY $w0
    %1:_(s32) = COPY $w1
    %2:_(s16) = G_TRUNC %0
    %3:_(s16) = G_TRUNC %1
    %6:_(s16) = G_FREEZE %2
    %4:_(s16) = G_AND %6, %3
    %5:_(s32) = G_ANYEXT %4
    $w0 = COPY %5
...

# and(freeze(trunc(x)), freeze(trunc(y))) -> trunc(and(freeze(x), freeze(y)))
---
name:            and_trunc_freeze_both
body:             |
  bb.0:
    liveins: $w0, $w1
    ; CHECK-LABEL: name: and_trunc_freeze_both
    ; CHECK: liveins: $w0, $w1
    ; CHECK-NEXT: {{  $}}
    ; CHECK-NEXT: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
    ; CHECK-NEXT: [[COPY1:%[0-9]+]]:_(s32) = COPY $w1
    ; CHECK-NEXT: [[FREEZE:%[0-9]+]]:_(s32) = G_FREEZE [[COPY]]
    ; CHECK-NEXT: [[FREEZE1:%[0-9]+]]:_(s32) = G_FREEZE [[COPY1]]
    ; CHECK-NEXT: [[AND:%[0-9]+]]:_(s32) = G_AND [[FREEZE]], [[FREEZE1]]
    ; CHECK-NEXT: $w0 = COPY [[AND]](s32)
    %0:_(s32) = COPY $w0
    %1:_(s32) = COPY $w1
    %2:_(s16) = G_TRUNC %0
    %3:_(s16) = G_TRUNC %1
    %6:_(s16) = G_FREEZE %2
    %7:_(s16) = G_FREEZE %3
    %4:_(s16) = G_AND %6, %7
    %5:_(s32) = G_ANYEXT %4
    $w0 = COPY %5
...

# The freeze fold is less important for G_OR and G_XOR, however it can still
# trigger.
---
name:            or_trunc_freeze
body:             |
  bb.0:
    liveins: $w0, $w1
    ; CHECK-LABEL: name: or_trunc_freeze
    ; CHECK: liveins: $w0, $w1
    ; CHECK-NEXT: {{  $}}
    ; CHECK-NEXT: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
    ; CHECK-NEXT: [[COPY1:%[0-9]+]]:_(s32) = COPY $w1
    ; CHECK-NEXT: [[FREEZE:%[0-9]+]]:_(s32) = G_FREEZE [[COPY]]
    ; CHECK-NEXT: [[OR:%[0-9]+]]:_(s32) = G_OR [[FREEZE]], [[COPY1]]
    ; CHECK-NEXT: $w0 = COPY [[OR]](s32)
    %0:_(s32) = COPY $w0
    %1:_(s32) = COPY $w1
    %2:_(s16) = G_TRUNC %0
    %3:_(s16) = G_TRUNC %1
    %6:_(s16) = G_FREEZE %2
    %4:_(s16) = G_OR %6, %3
    %5:_(s32) = G_ANYEXT %4
    $w0 = COPY %5
...
---
name:            xor_trunc_freeze_both
body:             |
  bb.0:
    liveins: $w0, $w1
    ; CHECK-LABEL: name: xor_trunc_freeze_both
    ; CHECK: liveins: $w0, $w1
    ; CHECK-NEXT: {{  $}}
    ; CHECK-NEXT: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
    ; CHECK-NEXT: [[COPY1:%[0-9]+]]:_(s32) = COPY $w1
    ; CHECK-NEXT: [[FREEZE:%[0-9]+]]:_(s32) = G_FREEZE [[COPY]]
    ; CHECK-NEXT: [[FREEZE1:%[0-9]+]]:_(s32) = G_FREEZE [[COPY1]]
    ; CHECK-NEXT: [[XOR:%[0-9]+]]:_(s32) = G_XOR [[FREEZE]], [[FREEZE1]]
    ; CHECK-NEXT: $w0 = COPY [[XOR]](s32)
    %0:_(s32) = COPY $w0
    %1:_(s32) = COPY $w1
    %2:_(s16) = G_TRUNC %0
    %3:_(s16) = G_TRUNC %1
    %6:_(s16) = G_FREEZE %2
    %7:_(s16) = G_FREEZE %3
    %4:_(s16) = G_XOR %6, %7
    %5:_(s32) = G_ANYEXT %4
    $w0 = COPY %5
...