# 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_ZEXT.
#
...
---
name: gpr
alignment: 4
legalized: true
tracksRegLiveness: true
body: |
bb.0:
liveins: $w0, $w1
; G_ASSERT_ZEXT should end up on a GPR.
; CHECK-LABEL: name: gpr
; CHECK: liveins: $w0, $w1
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: %copy:gpr(s32) = COPY $w0
; CHECK-NEXT: %copy_assert_zext:gpr(s32) = G_ASSERT_ZEXT %copy, 16
; CHECK-NEXT: $w1 = COPY %copy_assert_zext(s32)
; CHECK-NEXT: RET_ReallyLR implicit $w1
%copy:_(s32) = COPY $w0
%copy_assert_zext:_(s32) = G_ASSERT_ZEXT %copy(s32), 16
$w1 = COPY %copy_assert_zext(s32)
RET_ReallyLR implicit $w1
...
---
name: gpr_vector
alignment: 4
legalized: true
tracksRegLiveness: true
body: |
bb.0:
liveins: $x0, $x1
; G_ASSERT_ZEXT should end up on a GPR.
; CHECK-LABEL: name: gpr_vector
; CHECK: liveins: $x0, $x1
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: %copy:gpr(<2 x s32>) = COPY $x0
; CHECK-NEXT: %copy_assert_zext:gpr(<2 x s32>) = G_ASSERT_ZEXT %copy, 16
; CHECK-NEXT: $x1 = COPY %copy_assert_zext(<2 x s32>)
; CHECK-NEXT: RET_ReallyLR implicit $x1
%copy:_(<2 x s32>) = COPY $x0
%copy_assert_zext:_(<2 x s32>) = G_ASSERT_ZEXT %copy(<2 x s32>), 16
$x1 = COPY %copy_assert_zext(<2 x s32>)
RET_ReallyLR implicit $x1
...
---
name: fpr
alignment: 4
legalized: true
tracksRegLiveness: true
body: |
bb.0:
liveins: $s0, $s1
; G_ASSERT_ZEXT should end up on a FPR.
; CHECK-LABEL: name: fpr
; CHECK: liveins: $s0, $s1
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: %copy:fpr(s32) = COPY $s0
; CHECK-NEXT: %copy_assert_zext:fpr(s32) = G_ASSERT_ZEXT %copy, 16
; CHECK-NEXT: $s1 = COPY %copy_assert_zext(s32)
; CHECK-NEXT: RET_ReallyLR implicit $s1
%copy:_(s32) = COPY $s0
%copy_assert_zext:_(s32) = G_ASSERT_ZEXT %copy(s32), 16
$s1 = COPY %copy_assert_zext(s32)
RET_ReallyLR implicit $s1
...
---
name: fpr_vector
alignment: 4
legalized: true
tracksRegLiveness: true
body: |
bb.0:
liveins: $d0, $d1
; G_ASSERT_ZEXT should end up on a FPR.
; CHECK-LABEL: name: fpr_vector
; CHECK: liveins: $d0, $d1
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: %copy:fpr(<2 x s32>) = COPY $d0
; CHECK-NEXT: %copy_assert_zext:fpr(<2 x s32>) = G_ASSERT_ZEXT %copy, 16
; CHECK-NEXT: $d1 = COPY %copy_assert_zext(<2 x s32>)
; CHECK-NEXT: RET_ReallyLR implicit $d1
%copy:_(<2 x s32>) = COPY $d0
%copy_assert_zext:_(<2 x s32>) = G_ASSERT_ZEXT %copy(<2 x s32>), 16
$d1 = COPY %copy_assert_zext(<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-NEXT: {{ $}}
; CHECK-NEXT: %copy:fpr(s32) = COPY $s0
; CHECK-NEXT: %copy_assert_zext:fpr(s32) = G_ASSERT_ZEXT %copy, 16
; CHECK-NEXT: $w1 = COPY %copy_assert_zext(s32)
; CHECK-NEXT: RET_ReallyLR implicit $w1
%copy:_(s32) = COPY $s0
%copy_assert_zext:_(s32) = G_ASSERT_ZEXT %copy(s32), 16
$w1 = COPY %copy_assert_zext(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_ZEXT 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-NEXT: {{ $}}
; CHECK-NEXT: %ptr:gpr(p0) = COPY $x0
; CHECK-NEXT: %copy:fpr(s32) = COPY $s0
; CHECK-NEXT: %copy_assert_zext:fpr(s32) = G_ASSERT_ZEXT %copy, 16
; CHECK-NEXT: G_STORE %copy_assert_zext(s32), %ptr(p0) :: (store (s32))
; CHECK-NEXT: RET_ReallyLR
%ptr:_(p0) = COPY $x0
%copy:_(s32) = COPY $s0
%copy_assert_zext:_(s32) = G_ASSERT_ZEXT %copy(s32), 16
G_STORE %copy_assert_zext(s32), %ptr(p0) :: (store (s32))
RET_ReallyLR
...
---
name: fpr_feeding_select
alignment: 4
legalized: true
tracksRegLiveness: true
machineFunctionInfo: {}
body: |
bb.0:
liveins: $d0, $x1, $w0
; G_ASSERT_ZEXT and G_SELECT should both end up on FPRs.
; CHECK-LABEL: name: fpr_feeding_select
; CHECK: liveins: $d0, $x1, $w0
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: %w0:gpr(s32) = COPY $w0
; CHECK-NEXT: %fpr:fpr(s64) = COPY $d0
; CHECK-NEXT: %fpr_assert_zext:fpr(s64) = G_ASSERT_ZEXT %fpr, 32
; CHECK-NEXT: %gpr:gpr(s64) = COPY $x1
; CHECK-NEXT: [[COPY:%[0-9]+]]:fpr(s64) = COPY %gpr(s64)
; CHECK-NEXT: %select:fpr(s64) = G_SELECT %w0(s32), %fpr_assert_zext, [[COPY]]
; CHECK-NEXT: $d0 = COPY %select(s64)
; CHECK-NEXT: RET_ReallyLR implicit $d0
%w0:_(s32) = COPY $w0
%fpr:_(s64) = COPY $d0
%fpr_assert_zext:_(s64) = G_ASSERT_ZEXT %fpr, 32
%gpr:_(s64) = COPY $x1
%select:_(s64) = G_SELECT %w0, %fpr_assert_zext, %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-NEXT: successors: %bb.1(0x40000000), %bb.2(0x40000000)
; CHECK-NEXT: liveins: $s0, $w1
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: %copy1:fpr(s32) = COPY $s0
; CHECK-NEXT: %copy2:gpr(s32) = COPY $w1
; CHECK-NEXT: %copy_assert_zext:fpr(s32) = G_ASSERT_ZEXT %copy1, 16
; CHECK-NEXT: [[COPY:%[0-9]+]]:gpr(s32) = COPY %copy1(s32)
; CHECK-NEXT: %cmp:gpr(s32) = G_ICMP intpred(eq), [[COPY]](s32), %copy2
; CHECK-NEXT: G_BRCOND %cmp(s32), %bb.1
; CHECK-NEXT: G_BR %bb.1
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: bb.1:
; CHECK-NEXT: successors: %bb.2(0x80000000)
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: %bb1_val:gpr(s32) = COPY %copy2(s32)
; CHECK-NEXT: G_BR %bb.2
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: bb.2:
; CHECK-NEXT: successors: %bb.0(0x80000000)
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: %phi:fpr(s32) = G_PHI %copy_assert_zext(s32), %bb.0, %bb1_val(s32), %bb.1
; CHECK-NEXT: 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_zext:_(s32) = G_ASSERT_ZEXT %copy1(s32), 16
%cmp:_(s32) = G_ICMP intpred(eq), %copy1, %copy2
G_BRCOND %cmp, %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_zext, %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-NEXT: successors: %bb.1(0x40000000), %bb.2(0x40000000)
; CHECK-NEXT: liveins: $s0, $s1
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: %copy1:fpr(s32) = COPY $s0
; CHECK-NEXT: %copy2:fpr(s32) = COPY $s1
; CHECK-NEXT: [[COPY:%[0-9]+]]:gpr(s32) = COPY %copy1(s32)
; CHECK-NEXT: [[COPY1:%[0-9]+]]:gpr(s32) = COPY %copy2(s32)
; CHECK-NEXT: %cmp:gpr(s32) = G_ICMP intpred(eq), [[COPY]](s32), [[COPY1]]
; CHECK-NEXT: G_BRCOND %cmp(s32), %bb.1
; CHECK-NEXT: G_BR %bb.1
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: bb.1:
; CHECK-NEXT: successors: %bb.2(0x80000000)
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: %bb1_val:gpr(s32) = COPY %copy2(s32)
; CHECK-NEXT: G_BR %bb.2
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: bb.2:
; CHECK-NEXT: successors: %bb.0(0x80000000)
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: %phi:fpr(s32) = G_PHI %copy1(s32), %bb.0, %bb1_val(s32), %bb.1
; CHECK-NEXT: %assert_zext:fpr(s32) = G_ASSERT_ZEXT %phi, 16
; CHECK-NEXT: 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
G_BRCOND %cmp, %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_ZEXT should both end up on FPRs.
%phi:_(s32) = G_PHI %copy1, %bb.0, %bb1_val, %bb.1
%assert_zext:_(s32) = G_ASSERT_ZEXT %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-NEXT: successors: %bb.1(0x80000000)
; CHECK-NEXT: liveins: $w0, $w1
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: %copy:gpr(s32) = COPY $w0
; CHECK-NEXT: G_BR %bb.1
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: bb.1:
; CHECK-NEXT: %copy_assert_zext:gpr(s32) = G_ASSERT_ZEXT %copy, 16
; CHECK-NEXT: $w1 = COPY %copy_assert_zext(s32)
; CHECK-NEXT: 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_ZEXT should end up on a GPR.
%copy_assert_zext:_(s32) = G_ASSERT_ZEXT %copy(s32), 16
$w1 = COPY %copy_assert_zext
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-NEXT: successors: %bb.1(0x80000000)
; CHECK-NEXT: liveins: $s0, $s1
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: %copy:fpr(s32) = COPY $s0
; CHECK-NEXT: G_BR %bb.1
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: bb.1:
; CHECK-NEXT: %copy_assert_zext:fpr(s32) = G_ASSERT_ZEXT %copy, 16
; CHECK-NEXT: $s1 = COPY %copy_assert_zext(s32)
; CHECK-NEXT: 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_ZEXT should end up on a FPR.
%copy_assert_zext:_(s32) = G_ASSERT_ZEXT %copy(s32), 16
$s1 = COPY %copy_assert_zext
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-NEXT: successors: %bb.1(0x80000000)
; CHECK-NEXT: liveins: $s0, $s1
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: %copy:fpr(s32) = COPY $s0
; CHECK-NEXT: G_BR %bb.1
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: bb.1:
; CHECK-NEXT: successors: %bb.2(0x80000000)
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: %copy_assert_zext1:fpr(s32) = G_ASSERT_ZEXT %copy, 16
; CHECK-NEXT: G_BR %bb.2
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: bb.2:
; CHECK-NEXT: successors: %bb.0(0x80000000)
; CHECK-NEXT: {{ $}}
; CHECK-NEXT: %copy_assert_zext2:fpr(s32) = G_ASSERT_ZEXT %copy_assert_zext1, 16
; CHECK-NEXT: %copy_assert_zext3:fpr(s32) = G_ASSERT_ZEXT %copy_assert_zext2, 16
; CHECK-NEXT: 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_ZEXTs should end up on FPRs.
%copy_assert_zext1:_(s32) = G_ASSERT_ZEXT %copy(s32), 16
G_BR %bb.2
bb.2:
successors: %bb.0
%copy_assert_zext2:_(s32) = G_ASSERT_ZEXT %copy_assert_zext1(s32), 16
%copy_assert_zext3:_(s32) = G_ASSERT_ZEXT %copy_assert_zext2(s32), 16
G_BR %bb.0