llvm/llvm/test/CodeGen/AArch64/GlobalISel/combine-trunc.mir

# 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 --check-prefixes=CHECK,CHECK-PRE
# RUN: llc -o - -mtriple=aarch64-unknown-unknown -run-pass=aarch64-postlegalizer-combiner -verify-machineinstrs  %s | FileCheck %s --check-prefixes=CHECK,CHECK-POST

---
name:            test_combine_trunc_undef
legalized: true
body:             |
  bb.1:
    ; CHECK-LABEL: name: test_combine_trunc_undef
    ; CHECK: [[DEF:%[0-9]+]]:_(s32) = G_IMPLICIT_DEF
    ; CHECK-NEXT: $w0 = COPY [[DEF]](s32)
    %0:_(s64) = G_IMPLICIT_DEF
    %1:_(s32) = G_TRUNC %0(s64)
    $w0 = COPY %1(s32)
...
---
name:            test_combine_trunc_undef_vec
legalized: true
body:             |
  bb.1:
    ; CHECK-LABEL: name: test_combine_trunc_undef_vec
    ; CHECK: [[DEF:%[0-9]+]]:_(<2 x s32>) = G_IMPLICIT_DEF
    ; CHECK-NEXT: $x0 = COPY [[DEF]](<2 x s32>)
    %0:_(<2 x s64>) = G_IMPLICIT_DEF
    %1:_(<2 x s32>) = G_TRUNC %0(<2 x s64>)
    $x0 = COPY %1(<2 x s32>)
...
---
name:            test_combine_trunc_anyext_s32_s16
legalized: true
body:             |
  bb.1:
  liveins: $h0
    ; CHECK-PRE-LABEL: name: test_combine_trunc_anyext_s32_s16
    ; CHECK-PRE: liveins: $h0
    ; CHECK-PRE-NEXT: {{  $}}
    ; CHECK-PRE-NEXT: [[COPY:%[0-9]+]]:_(s16) = COPY $h0
    ; CHECK-PRE-NEXT: [[ANYEXT:%[0-9]+]]:_(s32) = G_ANYEXT [[COPY]](s16)
    ; CHECK-PRE-NEXT: $w0 = COPY [[ANYEXT]](s32)
    ;
    ; CHECK-POST-LABEL: name: test_combine_trunc_anyext_s32_s16
    ; CHECK-POST: liveins: $h0
    ; CHECK-POST-NEXT: {{  $}}
    ; CHECK-POST-NEXT: [[COPY:%[0-9]+]]:_(s16) = COPY $h0
    ; CHECK-POST-NEXT: [[ANYEXT:%[0-9]+]]:_(s64) = G_ANYEXT [[COPY]](s16)
    ; CHECK-POST-NEXT: [[TRUNC:%[0-9]+]]:_(s32) = G_TRUNC [[ANYEXT]](s64)
    ; CHECK-POST-NEXT: $w0 = COPY [[TRUNC]](s32)
    %0:_(s16) = COPY $h0
    %1:_(s64) = G_ANYEXT %0(s16)
    %2:_(s32) = G_TRUNC %1(s64)
    $w0 = COPY %2(s32)
...
---
name:            test_combine_trunc_anyext_s32_s16_vec
legalized: true
body:             |
  bb.1:
  liveins: $s0
    ; CHECK-PRE-LABEL: name: test_combine_trunc_anyext_s32_s16_vec
    ; CHECK-PRE: liveins: $s0
    ; CHECK-PRE-NEXT: {{  $}}
    ; CHECK-PRE-NEXT: [[COPY:%[0-9]+]]:_(<2 x s16>) = COPY $s0
    ; CHECK-PRE-NEXT: [[ANYEXT:%[0-9]+]]:_(<2 x s32>) = G_ANYEXT [[COPY]](<2 x s16>)
    ; CHECK-PRE-NEXT: $x0 = COPY [[ANYEXT]](<2 x s32>)
    ;
    ; CHECK-POST-LABEL: name: test_combine_trunc_anyext_s32_s16_vec
    ; CHECK-POST: liveins: $s0
    ; CHECK-POST-NEXT: {{  $}}
    ; CHECK-POST-NEXT: [[COPY:%[0-9]+]]:_(<2 x s16>) = COPY $s0
    ; CHECK-POST-NEXT: [[ANYEXT:%[0-9]+]]:_(<2 x s64>) = G_ANYEXT [[COPY]](<2 x s16>)
    ; CHECK-POST-NEXT: [[TRUNC:%[0-9]+]]:_(<2 x s32>) = G_TRUNC [[ANYEXT]](<2 x s64>)
    ; CHECK-POST-NEXT: $x0 = COPY [[TRUNC]](<2 x s32>)
    %0:_(<2 x s16>) = COPY $s0
    %1:_(<2 x s64>) = G_ANYEXT %0(<2 x s16>)
    %2:_(<2 x s32>) = G_TRUNC %1(<2 x s64>)
    $x0 = COPY %2(<2 x s32>)
...
---
name:            test_combine_trunc_sext_s32_s16
legalized: true
body:             |
  bb.1:
  liveins: $h0
    ; CHECK-PRE-LABEL: name: test_combine_trunc_sext_s32_s16
    ; CHECK-PRE: liveins: $h0
    ; CHECK-PRE-NEXT: {{  $}}
    ; CHECK-PRE-NEXT: [[COPY:%[0-9]+]]:_(s16) = COPY $h0
    ; CHECK-PRE-NEXT: [[SEXT:%[0-9]+]]:_(s32) = G_SEXT [[COPY]](s16)
    ; CHECK-PRE-NEXT: $w0 = COPY [[SEXT]](s32)
    ;
    ; CHECK-POST-LABEL: name: test_combine_trunc_sext_s32_s16
    ; CHECK-POST: liveins: $h0
    ; CHECK-POST-NEXT: {{  $}}
    ; CHECK-POST-NEXT: [[COPY:%[0-9]+]]:_(s16) = COPY $h0
    ; CHECK-POST-NEXT: [[SEXT:%[0-9]+]]:_(s64) = G_SEXT [[COPY]](s16)
    ; CHECK-POST-NEXT: [[TRUNC:%[0-9]+]]:_(s32) = G_TRUNC [[SEXT]](s64)
    ; CHECK-POST-NEXT: $w0 = COPY [[TRUNC]](s32)
    %0:_(s16) = COPY $h0
    %1:_(s64) = G_SEXT %0(s16)
    %2:_(s32) = G_TRUNC %1(s64)
    $w0 = COPY %2(s32)
...
---
name:            test_combine_trunc_zext_s32_s16
legalized: true
body:             |
  bb.1:
  liveins: $h0
    ; CHECK-PRE-LABEL: name: test_combine_trunc_zext_s32_s16
    ; CHECK-PRE: liveins: $h0
    ; CHECK-PRE-NEXT: {{  $}}
    ; CHECK-PRE-NEXT: [[COPY:%[0-9]+]]:_(s16) = COPY $h0
    ; CHECK-PRE-NEXT: [[ZEXT:%[0-9]+]]:_(s32) = G_ZEXT [[COPY]](s16)
    ; CHECK-PRE-NEXT: $w0 = COPY [[ZEXT]](s32)
    ;
    ; CHECK-POST-LABEL: name: test_combine_trunc_zext_s32_s16
    ; CHECK-POST: liveins: $h0
    ; CHECK-POST-NEXT: {{  $}}
    ; CHECK-POST-NEXT: [[COPY:%[0-9]+]]:_(s16) = COPY $h0
    ; CHECK-POST-NEXT: [[ZEXT:%[0-9]+]]:_(s64) = G_ZEXT [[COPY]](s16)
    ; CHECK-POST-NEXT: [[TRUNC:%[0-9]+]]:_(s32) = G_TRUNC [[ZEXT]](s64)
    ; CHECK-POST-NEXT: $w0 = COPY [[TRUNC]](s32)
    %0:_(s16) = COPY $h0
    %1:_(s64) = G_ZEXT %0(s16)
    %2:_(s32) = G_TRUNC %1(s64)
    $w0 = COPY %2(s32)
...
---
name:            test_combine_trunc_anyext_s32_s32
legalized: true
body:             |
  bb.1:
  liveins: $w0
    ; CHECK-PRE-LABEL: name: test_combine_trunc_anyext_s32_s32
    ; CHECK-PRE: liveins: $w0
    ; CHECK-PRE-NEXT: {{  $}}
    ; CHECK-PRE-NEXT: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
    ; CHECK-PRE-NEXT: $w0 = COPY [[COPY]](s32)
    ;
    ; CHECK-POST-LABEL: name: test_combine_trunc_anyext_s32_s32
    ; CHECK-POST: liveins: $w0
    ; CHECK-POST-NEXT: {{  $}}
    ; CHECK-POST-NEXT: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
    ; CHECK-POST-NEXT: [[ANYEXT:%[0-9]+]]:_(s64) = G_ANYEXT [[COPY]](s32)
    ; CHECK-POST-NEXT: [[TRUNC:%[0-9]+]]:_(s32) = G_TRUNC [[ANYEXT]](s64)
    ; CHECK-POST-NEXT: $w0 = COPY [[TRUNC]](s32)
    %0:_(s32) = COPY $w0
    %1:_(s64) = G_ANYEXT %0(s32)
    %2:_(s32) = G_TRUNC %1(s64)
    $w0 = COPY %2(s32)
...
---
name:            test_combine_trunc_anyext_s32_s64
legalized: true
body:             |
  bb.1:
  liveins: $x0
    ; CHECK-PRE-LABEL: name: test_combine_trunc_anyext_s32_s64
    ; CHECK-PRE: liveins: $x0
    ; CHECK-PRE-NEXT: {{  $}}
    ; CHECK-PRE-NEXT: [[COPY:%[0-9]+]]:_(s64) = COPY $x0
    ; CHECK-PRE-NEXT: [[TRUNC:%[0-9]+]]:_(s32) = G_TRUNC [[COPY]](s64)
    ; CHECK-PRE-NEXT: $w0 = COPY [[TRUNC]](s32)
    ;
    ; CHECK-POST-LABEL: name: test_combine_trunc_anyext_s32_s64
    ; CHECK-POST: liveins: $x0
    ; CHECK-POST-NEXT: {{  $}}
    ; CHECK-POST-NEXT: [[COPY:%[0-9]+]]:_(s64) = COPY $x0
    ; CHECK-POST-NEXT: [[ANYEXT:%[0-9]+]]:_(s128) = G_ANYEXT [[COPY]](s64)
    ; CHECK-POST-NEXT: [[TRUNC:%[0-9]+]]:_(s32) = G_TRUNC [[ANYEXT]](s128)
    ; CHECK-POST-NEXT: $w0 = COPY [[TRUNC]](s32)
    %0:_(s64) = COPY $x0
    %1:_(s128) = G_ANYEXT %0(s64)
    %2:_(s32) = G_TRUNC %1(s128)
    $w0 = COPY %2(s32)
...
---
name:            test_combine_trunc_shl_s32_by_2
legalized: true
body:             |
  bb.1:
  liveins: $w0
    ; CHECK-PRE-LABEL: name: test_combine_trunc_shl_s32_by_2
    ; CHECK-PRE: liveins: $w0
    ; CHECK-PRE-NEXT: {{  $}}
    ; CHECK-PRE-NEXT: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
    ; CHECK-PRE-NEXT: [[C:%[0-9]+]]:_(s32) = G_CONSTANT i32 2
    ; CHECK-PRE-NEXT: [[TRUNC:%[0-9]+]]:_(s16) = G_TRUNC [[COPY]](s32)
    ; CHECK-PRE-NEXT: [[SHL:%[0-9]+]]:_(s16) = G_SHL [[TRUNC]], [[C]](s32)
    ; CHECK-PRE-NEXT: $h0 = COPY [[SHL]](s16)
    ;
    ; CHECK-POST-LABEL: name: test_combine_trunc_shl_s32_by_2
    ; CHECK-POST: liveins: $w0
    ; CHECK-POST-NEXT: {{  $}}
    ; CHECK-POST-NEXT: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
    ; CHECK-POST-NEXT: [[C:%[0-9]+]]:_(s32) = G_CONSTANT i32 2
    ; CHECK-POST-NEXT: [[SHL:%[0-9]+]]:_(s32) = G_SHL [[COPY]], [[C]](s32)
    ; CHECK-POST-NEXT: [[TRUNC:%[0-9]+]]:_(s16) = G_TRUNC [[SHL]](s32)
    ; CHECK-POST-NEXT: $h0 = COPY [[TRUNC]](s16)
    %0:_(s32) = COPY $w0
    %1:_(s32) = G_CONSTANT i32 2
    %2:_(s32) = G_SHL %0(s32), %1(s32)
    %3:_(s16) = G_TRUNC %2(s32)
    $h0 = COPY %3(s16)
...
---
name:            test_combine_trunc_shl_s32_by_17
legalized: true
body:             |
  bb.1:
  liveins: $w0
    ; CHECK-LABEL: name: test_combine_trunc_shl_s32_by_17
    ; CHECK: liveins: $w0
    ; CHECK-NEXT: {{  $}}
    ; CHECK-NEXT: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
    ; CHECK-NEXT: [[C:%[0-9]+]]:_(s32) = G_CONSTANT i32 17
    ; CHECK-NEXT: [[SHL:%[0-9]+]]:_(s32) = G_SHL [[COPY]], [[C]](s32)
    ; CHECK-NEXT: [[TRUNC:%[0-9]+]]:_(s16) = G_TRUNC [[SHL]](s32)
    ; CHECK-NEXT: $h0 = COPY [[TRUNC]](s16)
    %0:_(s32) = COPY $w0
    %1:_(s32) = G_CONSTANT i32 17
    %2:_(s32) = G_SHL %0(s32), %1(s32)
    %3:_(s16) = G_TRUNC %2(s32)
    $h0 = COPY %3(s16)
...
---
name:            test_combine_trunc_multi_use
legalized: true
body:             |
  bb.1:
  liveins: $w0
    ; CHECK-LABEL: name: test_combine_trunc_multi_use
    ; CHECK: liveins: $w0
    ; CHECK-NEXT: {{  $}}
    ; CHECK-NEXT: [[COPY:%[0-9]+]]:_(s32) = COPY $w0
    ; CHECK-NEXT: [[ZEXT:%[0-9]+]]:_(s64) = G_ZEXT [[COPY]](s32)
    ; CHECK-NEXT: [[TRUNC:%[0-9]+]]:_(s16) = G_TRUNC [[ZEXT]](s64)
    ; CHECK-NEXT: $h0 = COPY [[TRUNC]](s16)
    ; CHECK-NEXT: $x0 = COPY [[ZEXT]](s64)
    %0:_(s32) = COPY $w0
    %2:_(s64) = G_ZEXT %0(s32)
    %3:_(s16) = G_TRUNC %2(s64)
    $h0 = COPY %3(s16)
    $x0 = COPY %2(s64)
...