combine-icmp-to-lhs-known-bits.mir | Explore in Territory

# NOTE: Assertions have been autogenerated by utils/update_mir_test_checks.py

# RUN: llc -mtriple aarch64 -run-pass=aarch64-prelegalizer-combiner --aarch64prelegalizercombiner-only-enable-rule="icmp_to_lhs_known_bits" -global-isel -verify-machineinstrs %s -o - | FileCheck %s
# REQUIRES: asserts

...
---
name:            apply_ne
alignment:       4
tracksRegLiveness: true
machineFunctionInfo: {}
body:             |
  bb.0:
    liveins: $w0
    ; CHECK-LABEL: name: apply_ne
    ; CHECK: liveins: $w0
    ; CHECK: %x:_(s32) = COPY $w0
    ; CHECK: %one:_(s32) = G_CONSTANT i32 1
    ; CHECK: %known_zero_or_one:_(s32) = G_AND %x, %one
    ; CHECK: %cmp:_(s1) = G_TRUNC %known_zero_or_one(s32)
    ; CHECK: %ext:_(s32) = G_ZEXT %cmp(s1)
    ; CHECK: $w0 = COPY %ext(s32)
    ; CHECK: RET_ReallyLR implicit $w0
    %x:_(s32) = COPY $w0
    %one:_(s32) = G_CONSTANT i32 1
    %known_zero_or_one:_(s32) = G_AND %x, %one
    %zero:_(s32) = G_CONSTANT i32 0
    %cmp:_(s1) = G_ICMP intpred(ne), %known_zero_or_one(s32), %zero
    %ext:_(s32) = G_ZEXT %cmp(s1)
    $w0 = COPY %ext(s32)
    RET_ReallyLR implicit $w0

...
---
name:            apply_eq
alignment:       4
tracksRegLiveness: true
machineFunctionInfo: {}
body:             |
  bb.0:
    liveins: $w0
    ; CHECK-LABEL: name: apply_eq
    ; CHECK: liveins: $w0
    ; CHECK: %x:_(s32) = COPY $w0
    ; CHECK: %one:_(s32) = G_CONSTANT i32 1
    ; CHECK: %known_zero_or_one:_(s32) = G_AND %x, %one
    ; CHECK: %cmp:_(s1) = G_TRUNC %known_zero_or_one(s32)
    ; CHECK: %ext:_(s32) = G_ZEXT %cmp(s1)
    ; CHECK: $w0 = COPY %ext(s32)
    ; CHECK: RET_ReallyLR implicit $w0
    %x:_(s32) = COPY $w0
    %one:_(s32) = G_CONSTANT i32 1
    %known_zero_or_one:_(s32) = G_AND %x, %one
    %cmp:_(s1) = G_ICMP intpred(eq), %known_zero_or_one(s32), %one
    %ext:_(s32) = G_ZEXT %cmp(s1)
    $w0 = COPY %ext(s32)
    RET_ReallyLR implicit $w0

...
---
name:            dont_apply_wrong_cst_eq
alignment:       4
tracksRegLiveness: true
machineFunctionInfo: {}
body:             |
  bb.0:
    liveins: $w0
    ; Wrong constant on the RHS of the compare.

    ; CHECK-LABEL: name: dont_apply_wrong_cst_eq
    ; CHECK: liveins: $w0
    ; CHECK: %x:_(s32) = COPY $w0
    ; CHECK: %one:_(s32) = G_CONSTANT i32 1
    ; CHECK: %known_zero_or_one:_(s32) = G_AND %x, %one
    ; CHECK: %wrong_cst:_(s32) = G_CONSTANT i32 10
    ; CHECK: %cmp:_(s1) = G_ICMP intpred(eq), %known_zero_or_one(s32), %wrong_cst
    ; CHECK: %ext:_(s32) = G_ZEXT %cmp(s1)
    ; CHECK: $w0 = COPY %ext(s32)
    ; CHECK: RET_ReallyLR implicit $w0
    %x:_(s32) = COPY $w0
    %one:_(s32) = G_CONSTANT i32 1
    %known_zero_or_one:_(s32) = G_AND %x, %one
    %wrong_cst:_(s32) = G_CONSTANT i32 10
    %cmp:_(s1) = G_ICMP intpred(eq), %known_zero_or_one(s32), %wrong_cst
    %ext:_(s32) = G_ZEXT %cmp(s1)
    $w0 = COPY %ext(s32)
    RET_ReallyLR implicit $w0

...
---
name:            dont_apply_wrong_cst_ne
alignment:       4
tracksRegLiveness: true
machineFunctionInfo: {}
body:             |
  bb.0:
    liveins: $w0
    ; Wrong constant on the RHS of the compare.

    ; CHECK-LABEL: name: dont_apply_wrong_cst_ne
    ; CHECK: liveins: $w0
    ; CHECK: %x:_(s32) = COPY $w0
    ; CHECK: %one:_(s32) = G_CONSTANT i32 1
    ; CHECK: %known_zero_or_one:_(s32) = G_AND %x, %one
    ; CHECK: %wrong_cst:_(s32) = G_CONSTANT i32 10
    ; CHECK: %cmp:_(s1) = G_ICMP intpred(ne), %known_zero_or_one(s32), %wrong_cst
    ; CHECK: %ext:_(s32) = G_ZEXT %cmp(s1)
    ; CHECK: $w0 = COPY %ext(s32)
    ; CHECK: RET_ReallyLR implicit $w0
    %x:_(s32) = COPY $w0
    %one:_(s32) = G_CONSTANT i32 1
    %known_zero_or_one:_(s32) = G_AND %x, %one
    %wrong_cst:_(s32) = G_CONSTANT i32 10
    %cmp:_(s1) = G_ICMP intpred(ne), %known_zero_or_one(s32), %wrong_cst
    %ext:_(s32) = G_ZEXT %cmp(s1)
    $w0 = COPY %ext(s32)
    RET_ReallyLR implicit $w0

...
---
name:            dont_apply_vector
alignment:       4
tracksRegLiveness: true
machineFunctionInfo: {}
body:             |
  bb.0:
    liveins: $x0
    ; True is -1 for vectors on AArch64 so we don't want to combine.

    ; CHECK-LABEL: name: dont_apply_vector
    ; CHECK: liveins: $x0
    ; CHECK-NEXT: {{  $}}
    ; CHECK-NEXT: %x:_(<2 x s32>) = COPY $x0
    ; CHECK-NEXT: %one:_(s32) = G_CONSTANT i32 1
    ; CHECK-NEXT: %one_vec:_(<2 x s32>) = G_BUILD_VECTOR %one(s32), %one(s32)
    ; CHECK-NEXT: %vec_and:_(<2 x s32>) = G_AND %x, %one_vec
    ; CHECK-NEXT: %zero:_(s32) = G_CONSTANT i32 0
    ; CHECK-NEXT: %zero64:_(s64) = G_CONSTANT i64 0
    ; CHECK-NEXT: %zero_vec:_(<2 x s32>) = G_BUILD_VECTOR %zero(s32), %zero(s32)
    ; CHECK-NEXT: %cmp:_(<2 x s1>) = G_ICMP intpred(ne), %vec_and(<2 x s32>), %zero_vec
    ; CHECK-NEXT: %elt:_(s1) = G_EXTRACT_VECTOR_ELT %cmp(<2 x s1>), %zero64(s64)
    ; CHECK-NEXT: %ext:_(s32) = G_ZEXT %elt(s1)
    ; CHECK-NEXT: $w0 = COPY %ext(s32)
    ; CHECK-NEXT: RET_ReallyLR implicit $w0
    %x:_(<2 x s32>) = COPY $x0
    %one:_(s32) = G_CONSTANT i32 1
    %one_vec:_(<2 x s32>) = G_BUILD_VECTOR %one, %one
    %vec_and:_(<2 x s32>) = G_AND %x, %one_vec
    %zero:_(s32) = G_CONSTANT i32 0
    %zero64:_(s64) = G_CONSTANT i64 0
    %zero_vec:_(<2 x s32>) = G_BUILD_VECTOR %zero, %zero
    %cmp:_(<2 x s1>) = G_ICMP intpred(ne), %vec_and(<2 x s32>), %zero_vec
    %elt:_(s1) = G_EXTRACT_VECTOR_ELT %cmp, %zero64
    %ext:_(s32) = G_ZEXT %elt(s1)
    $w0 = COPY %ext(s32)
    RET_ReallyLR implicit $w0

...
---
name:            apply_no_zext_or_trunc
alignment:       4
tracksRegLiveness: true
machineFunctionInfo: {}
body:             |
  bb.0:
    liveins: $w0
    ; CHECK-LABEL: name: apply_no_zext_or_trunc
    ; CHECK: liveins: $w0
    ; CHECK: %x:_(s32) = COPY $w0
    ; CHECK: %one:_(s32) = G_CONSTANT i32 1
    ; CHECK: %known_zero_or_one:_(s32) = G_AND %x, %one
    ; CHECK: %cmp:_(s32) = COPY %known_zero_or_one(s32)
    ; CHECK: $w0 = COPY %cmp(s32)
    ; CHECK: RET_ReallyLR implicit $w0
    %x:_(s32) = COPY $w0
    %one:_(s32) = G_CONSTANT i32 1
    %known_zero_or_one:_(s32) = G_AND %x, %one
    %zero:_(s32) = G_CONSTANT i32 0
    %cmp:_(s32) = G_ICMP intpred(ne), %known_zero_or_one(s32), %zero
    $w0 = COPY %cmp(s32)
    RET_ReallyLR implicit $w0

...
---
name:            apply_wide_cmp
alignment:       4
tracksRegLiveness: true
machineFunctionInfo: {}
body:             |
  bb.0:
    liveins: $w0
    ; CHECK-LABEL: name: apply_wide_cmp
    ; CHECK: liveins: $w0
    ; CHECK: %x:_(s64) = COPY $x0
    ; CHECK: %one:_(s64) = G_CONSTANT i64 1
    ; CHECK: %known_zero_or_one:_(s64) = G_AND %x, %one
    ; CHECK: %cmp:_(s64) = COPY %known_zero_or_one(s64)
    ; CHECK: %trunc:_(s32) = G_TRUNC %cmp(s64)
    ; CHECK: $w0 = COPY %trunc(s32)
    ; CHECK: RET_ReallyLR implicit $w0
    %x:_(s64) = COPY $x0
    %one:_(s64) = G_CONSTANT i64 1
    %known_zero_or_one:_(s64) = G_AND %x, %one
    %zero:_(s64) = G_CONSTANT i64 0
    %cmp:_(s64) = G_ICMP intpred(ne), %known_zero_or_one(s64), %zero
    %trunc:_(s32) = G_TRUNC %cmp
    $w0 = COPY %trunc(s32)
    RET_ReallyLR implicit $w0

...
---
name:            apply_narrow_lhs
alignment:       4
tracksRegLiveness: true
machineFunctionInfo: {}
body:             |
  bb.0:
    liveins: $w0
    ; CHECK-LABEL: name: apply_narrow_lhs
    ; CHECK: liveins: $w0
    ; CHECK: %x:_(s32) = COPY $w0
    ; CHECK: %one:_(s32) = G_CONSTANT i32 1
    ; CHECK: %known_zero_or_one:_(s32) = G_AND %x, %one
    ; CHECK: %cmp:_(s64) = G_ZEXT %known_zero_or_one(s32)
    ; CHECK: $x0 = COPY %cmp(s64)
    ; CHECK: RET_ReallyLR implicit $x0
    %x:_(s32) = COPY $w0
    %one:_(s32) = G_CONSTANT i32 1
    %known_zero_or_one:_(s32) = G_AND %x, %one
    %zero:_(s32) = G_CONSTANT i32 0
    %cmp:_(s64) = G_ICMP intpred(ne), %known_zero_or_one(s32), %zero
    $x0 = COPY %cmp(s64)
    RET_ReallyLR implicit $x0
llvm/llvm/test/CodeGen/AArch64/GlobalISel/combine-icmp-to-lhs-known-bits.mir
