test/CodeGen/AArch64/GlobalISel/regbank-assert-sext.mir

# NOTE: Assertions have been autogenerated by utils/update_mir_test_checks.py
# RUN: llc -mtriple=aarch64 -run-pass=regbankselect -verify-machineinstrs %s -o - | FileCheck %s
#
# Verify register banks for G_ASSERT_SEXT.
#

...
---
name:            gpr
alignment:       4
legalized:       true
tracksRegLiveness: true
body:             |
  bb.0:
    liveins: $w0, $w1

    ; G_ASSERT_SEXT should end up on a GPR.

    ; CHECK-LABEL: name: gpr
    ; CHECK: liveins: $w0, $w1
    ; CHECK: %copy:gpr(s32) = COPY $w0
    ; CHECK: %copy_assert_sext:gpr(s32) = G_ASSERT_SEXT %copy, 16
    ; CHECK: $w1 = COPY %copy_assert_sext(s32)
    ; CHECK: RET_ReallyLR implicit $w1
    %copy:_(s32) = COPY $w0
    %copy_assert_sext:_(s32) = G_ASSERT_SEXT %copy(s32), 16
    $w1 = COPY %copy_assert_sext(s32)
    RET_ReallyLR implicit $w1

...
---
name:            gpr_vector
alignment:       4
legalized:       true
tracksRegLiveness: true
body:             |
  bb.0:
    liveins: $x0, $x1

    ; G_ASSERT_SEXT should end up on a GPR.

    ; CHECK-LABEL: name: gpr_vector
    ; CHECK: liveins: $x0, $x1
    ; CHECK: %copy:gpr(<2 x s32>) = COPY $x0
    ; CHECK: %copy_assert_sext:gpr(<2 x s32>) = G_ASSERT_SEXT %copy, 16
    ; CHECK: $x1 = COPY %copy_assert_sext(<2 x s32>)
    ; CHECK: RET_ReallyLR implicit $x1
    %copy:_(<2 x s32>) = COPY $x0
    %copy_assert_sext:_(<2 x s32>) = G_ASSERT_SEXT %copy(<2 x s32>), 16
    $x1 = COPY %copy_assert_sext(<2 x s32>)
    RET_ReallyLR implicit $x1

...
---
name:            fpr
alignment:       4
legalized:       true
tracksRegLiveness: true
body:             |
  bb.0:
    liveins: $s0, $s1

    ; G_ASSERT_SEXT should end up on a FPR.

    ; CHECK-LABEL: name: fpr
    ; CHECK: liveins: $s0, $s1
    ; CHECK: %copy:fpr(s32) = COPY $s0
    ; CHECK: %copy_assert_sext:fpr(s32) = G_ASSERT_SEXT %copy, 16
    ; CHECK: $s1 = COPY %copy_assert_sext(s32)
    ; CHECK: RET_ReallyLR implicit $s1
    %copy:_(s32) = COPY $s0
    %copy_assert_sext:_(s32) = G_ASSERT_SEXT %copy(s32), 16
    $s1 = COPY %copy_assert_sext(s32)
    RET_ReallyLR implicit $s1

...
---
name:            fpr_vector
alignment:       4
legalized:       true
tracksRegLiveness: true
body:             |
  bb.0:
    liveins: $d0, $d1

    ; G_ASSERT_SEXT should end up on a FPR.

    ; CHECK-LABEL: name: fpr_vector
    ; CHECK: liveins: $d0, $d1
    ; CHECK: %copy:fpr(<2 x s32>) = COPY $d0
    ; CHECK: %copy_assert_sext:fpr(<2 x s32>) = G_ASSERT_SEXT %copy, 16
    ; CHECK: $d1 = COPY %copy_assert_sext(<2 x s32>)
    ; CHECK: RET_ReallyLR implicit $d1
    %copy:_(<2 x s32>) = COPY $d0
    %copy_assert_sext:_(<2 x s32>) = G_ASSERT_SEXT %copy(<2 x s32>), 16
    $d1 = COPY %copy_assert_sext(<2 x s32>)
    RET_ReallyLR implicit $d1

...
---
name:            in_between_cross_bank_copy
alignment:       4
legalized:       true
tracksRegLiveness: true
body:             |
  bb.0:
    liveins: $s0, $w1

    ; CHECK-LABEL: name: in_between_cross_bank_copy
    ; CHECK: liveins: $s0, $w1
    ; CHECK: %copy:fpr(s32) = COPY $s0
    ; CHECK: %copy_assert_sext:fpr(s32) = G_ASSERT_SEXT %copy, 16
    ; CHECK: $w1 = COPY %copy_assert_sext(s32)
    ; CHECK: RET_ReallyLR implicit $w1
    %copy:_(s32) = COPY $s0
    %copy_assert_sext:_(s32) = G_ASSERT_SEXT %copy(s32), 16
    $w1 = COPY %copy_assert_sext(s32)
    RET_ReallyLR implicit $w1

...
---
name:            fpr_feeding_store
alignment:       4
legalized:       true
tracksRegLiveness: true
body:             |
  bb.0:
    liveins: $x0, $s0, $s1

    ; The G_ASSERT_SEXT should end up on a FPR, and there should be no copy
    ; between it and the G_STORE.

    ; CHECK-LABEL: name: fpr_feeding_store
    ; CHECK: liveins: $x0, $s0, $s1
    ; CHECK: %ptr:gpr(p0) = COPY $x0
    ; CHECK: %copy:fpr(s32) = COPY $s0
    ; CHECK: %copy_assert_sext:fpr(s32) = G_ASSERT_SEXT %copy, 16
    ; CHECK: G_STORE %copy_assert_sext(s32), %ptr(p0) :: (store 4)
    ; CHECK: RET_ReallyLR
    %ptr:_(p0) = COPY $x0
    %copy:_(s32) = COPY $s0
    %copy_assert_sext:_(s32) = G_ASSERT_SEXT %copy(s32), 16
    G_STORE %copy_assert_sext(s32), %ptr(p0) :: (store 4)
    RET_ReallyLR

...
---
name:            fpr_feeding_select
alignment:       4
legalized:       true
tracksRegLiveness: true
machineFunctionInfo: {}
body:             |
  bb.0:
    liveins: $d0, $x1, $w0

    ; G_ASSERT_SEXT and G_SELECT should both end up on FPRs.

    ; CHECK-LABEL: name: fpr_feeding_select
    ; CHECK: liveins: $d0, $x1, $w0
    ; CHECK: %w0:gpr(s32) = COPY $w0
    ; CHECK: %cond:gpr(s1) = G_TRUNC %w0(s32)
    ; CHECK: %fpr:fpr(s64) = COPY $d0
    ; CHECK: %fpr_assert_sext:fpr(s64) = G_ASSERT_SEXT %fpr, 32
    ; CHECK: %gpr:gpr(s64) = COPY $x1
    ; CHECK: [[COPY:%[0-9]+]]:fpr(s64) = COPY %gpr(s64)
    ; CHECK: %select:fpr(s64) = G_SELECT %cond(s1), %fpr_assert_sext, [[COPY]]
    ; CHECK: $d0 = COPY %select(s64)
    ; CHECK: RET_ReallyLR implicit $d0
    %w0:_(s32) = COPY $w0
    %cond:_(s1) = G_TRUNC %w0(s32)
    %fpr:_(s64) = COPY $d0
    %fpr_assert_sext:_(s64) = G_ASSERT_SEXT %fpr, 32
    %gpr:_(s64) = COPY $x1
    %select:_(s64) = G_SELECT %cond(s1), %fpr_assert_sext, %gpr
    $d0 = COPY %select(s64)
    RET_ReallyLR implicit $d0

...
---
name:            fpr_feeding_phi
alignment:       4
legalized:       true
tracksRegLiveness: true
body:             |
  ; CHECK-LABEL: name: fpr_feeding_phi
  ; CHECK: bb.0:
  ; CHECK:   successors: %bb.1(0x40000000), %bb.2(0x40000000)
  ; CHECK:   liveins: $s0, $w1
  ; CHECK:   %copy1:fpr(s32) = COPY $s0
  ; CHECK:   %copy2:gpr(s32) = COPY $w1
  ; CHECK:   %copy_assert_sext:fpr(s32) = G_ASSERT_SEXT %copy1, 16
  ; CHECK:   [[COPY:%[0-9]+]]:gpr(s32) = COPY %copy1(s32)
  ; CHECK:   %cmp:gpr(s32) = G_ICMP intpred(eq), [[COPY]](s32), %copy2
  ; CHECK:   %cmp_trunc:gpr(s1) = G_TRUNC %cmp(s32)
  ; CHECK:   G_BRCOND %cmp_trunc(s1), %bb.1
  ; CHECK:   G_BR %bb.1
  ; CHECK: bb.1:
  ; CHECK:   successors: %bb.2(0x80000000)
  ; CHECK:   %bb1_val:gpr(s32) = COPY %copy2(s32)
  ; CHECK:   G_BR %bb.2
  ; CHECK: bb.2:
  ; CHECK:   successors: %bb.0(0x80000000)
  ; CHECK:   %phi:fpr(s32) = G_PHI %copy_assert_sext(s32), %bb.0, %bb1_val(s32), %bb.1
  ; CHECK:   G_BR %bb.0
  bb.0:
    successors: %bb.1, %bb.2
    liveins: $s0, $w1
    %copy1:_(s32) = COPY $s0
    %copy2:_(s32) = COPY $w1

    ; This should produce a FPR.
    %copy_assert_sext:_(s32) = G_ASSERT_SEXT %copy1(s32), 16

    %cmp:_(s32) = G_ICMP intpred(eq), %copy1, %copy2
    %cmp_trunc:_(s1) = G_TRUNC %cmp
    G_BRCOND %cmp_trunc, %bb.1
    G_BR %bb.1
  bb.1:
    successors: %bb.2
    %bb1_val:_(s32) = COPY %copy2
    G_BR %bb.2
  bb.2:
    successors: %bb.0
    ; This should produce a FPR.
    %phi:_(s32) = G_PHI %copy_assert_sext, %bb.0, %bb1_val, %bb.1
    G_BR %bb.0

...
---
name:            fed_by_fpr_phi
alignment:       4
legalized:       true
tracksRegLiveness: true
body:             |
  ; CHECK-LABEL: name: fed_by_fpr_phi
  ; CHECK: bb.0:
  ; CHECK:   successors: %bb.1(0x40000000), %bb.2(0x40000000)
  ; CHECK:   liveins: $s0, $s1
  ; CHECK:   %copy1:fpr(s32) = COPY $s0
  ; CHECK:   %copy2:fpr(s32) = COPY $s1
  ; CHECK:   [[COPY:%[0-9]+]]:gpr(s32) = COPY %copy1(s32)
  ; CHECK:   [[COPY1:%[0-9]+]]:gpr(s32) = COPY %copy2(s32)
  ; CHECK:   %cmp:gpr(s32) = G_ICMP intpred(eq), [[COPY]](s32), [[COPY1]]
  ; CHECK:   %cmp_trunc:gpr(s1) = G_TRUNC %cmp(s32)
  ; CHECK:   G_BRCOND %cmp_trunc(s1), %bb.1
  ; CHECK:   G_BR %bb.1
  ; CHECK: bb.1:
  ; CHECK:   successors: %bb.2(0x80000000)
  ; CHECK:   %bb1_val:gpr(s32) = COPY %copy2(s32)
  ; CHECK:   G_BR %bb.2
  ; CHECK: bb.2:
  ; CHECK:   successors: %bb.0(0x80000000)
  ; CHECK:   %phi:fpr(s32) = G_PHI %copy1(s32), %bb.0, %bb1_val(s32), %bb.1
  ; CHECK:   %assert_sext:fpr(s32) = G_ASSERT_SEXT %phi, 16
  ; CHECK:   G_BR %bb.0
  bb.0:
    successors: %bb.1, %bb.2
    liveins: $s0, $s1
    %copy1:_(s32) = COPY $s0
    %copy2:_(s32) = COPY $s1
    %cmp:_(s32) = G_ICMP intpred(eq), %copy1, %copy2
    %cmp_trunc:_(s1) = G_TRUNC %cmp
    G_BRCOND %cmp_trunc, %bb.1
    G_BR %bb.1
  bb.1:
    successors: %bb.2
    %bb1_val:_(s32) = COPY %copy2
    G_BR %bb.2
  bb.2:
    successors: %bb.0
    ; The G_PHI and G_ASSERT_SEXT should both end up on FPRs.
    %phi:_(s32) = G_PHI %copy1, %bb.0, %bb1_val, %bb.1
    %assert_sext:_(s32) = G_ASSERT_SEXT %phi(s32), 16
    G_BR %bb.0

...
---
name:            different_blocks_gpr
alignment:       4
legalized:       true
tracksRegLiveness: true
body:             |
  ; CHECK-LABEL: name: different_blocks_gpr
  ; CHECK: bb.0:
  ; CHECK:   successors: %bb.1(0x80000000)
  ; CHECK:   liveins: $w0, $w1
  ; CHECK:   %copy:gpr(s32) = COPY $w0
  ; CHECK:   G_BR %bb.1
  ; CHECK: bb.1:
  ; CHECK:   %copy_assert_sext:gpr(s32) = G_ASSERT_SEXT %copy, 16
  ; CHECK:   $w1 = COPY %copy_assert_sext(s32)
  ; CHECK:   RET_ReallyLR implicit $w1
  bb.0:
    successors: %bb.1
    liveins: $w0, $w1
    %copy:_(s32) = COPY $w0
    G_BR %bb.1
  bb.1:
    ; The G_ASSERT_SEXT should end up on a GPR.
    %copy_assert_sext:_(s32) = G_ASSERT_SEXT %copy(s32), 16
    $w1 = COPY %copy_assert_sext
    RET_ReallyLR implicit $w1

...
---
name:            different_blocks_fpr
alignment:       4
legalized:       true
tracksRegLiveness: true
body:             |
  ; CHECK-LABEL: name: different_blocks_fpr
  ; CHECK: bb.0:
  ; CHECK:   successors: %bb.1(0x80000000)
  ; CHECK:   liveins: $s0, $s1
  ; CHECK:   %copy:fpr(s32) = COPY $s0
  ; CHECK:   G_BR %bb.1
  ; CHECK: bb.1:
  ; CHECK:   %copy_assert_sext:fpr(s32) = G_ASSERT_SEXT %copy, 16
  ; CHECK:   $s1 = COPY %copy_assert_sext(s32)
  ; CHECK:   RET_ReallyLR implicit $s1
  bb.0:
    successors: %bb.1
    liveins: $s0, $s1
    %copy:_(s32) = COPY $s0
    G_BR %bb.1
  bb.1:
    ; The G_ASSERT_SEXT should end up on a FPR.
    %copy_assert_sext:_(s32) = G_ASSERT_SEXT %copy(s32), 16
    $s1 = COPY %copy_assert_sext
    RET_ReallyLR implicit $s1


...
---
name:            different_blocks_fpr_backedge
alignment:       4
legalized:       true
tracksRegLiveness: true
body:             |
  ; CHECK-LABEL: name: different_blocks_fpr_backedge
  ; CHECK: bb.0:
  ; CHECK:   successors: %bb.1(0x80000000)
  ; CHECK:   liveins: $s0, $s1
  ; CHECK:   %copy:fpr(s32) = COPY $s0
  ; CHECK:   G_BR %bb.1
  ; CHECK: bb.1:
  ; CHECK:   successors: %bb.2(0x80000000)
  ; CHECK:   %copy_assert_sext1:fpr(s32) = G_ASSERT_SEXT %copy, 16
  ; CHECK:   G_BR %bb.2
  ; CHECK: bb.2:
  ; CHECK:   successors: %bb.0(0x80000000)
  ; CHECK:   %copy_assert_sext2:fpr(s32) = G_ASSERT_SEXT %copy_assert_sext1, 16
  ; CHECK:   %copy_assert_sext3:fpr(s32) = G_ASSERT_SEXT %copy_assert_sext2, 16
  ; CHECK:   G_BR %bb.0
  bb.0:
    successors: %bb.1
    liveins: $s0, $s1
    %copy:_(s32) = COPY $s0
    G_BR %bb.1
  bb.1:
    successors: %bb.2
    ; All of the G_ASSERT_SEXTs should end up on FPRs.
    %copy_assert_sext1:_(s32) = G_ASSERT_SEXT %copy(s32), 16
    G_BR %bb.2
  bb.2:
    successors: %bb.0
    %copy_assert_sext2:_(s32) = G_ASSERT_SEXT %copy_assert_sext1(s32), 16
    %copy_assert_sext3:_(s32) = G_ASSERT_SEXT %copy_assert_sext2(s32), 16
    G_BR %bb.0