45 lines
1.8 KiB
LLVM
45 lines
1.8 KiB
LLVM
; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
|
|
; RUN: llc -mtriple=aarch64 -mattr=+sve < %s | FileCheck %s
|
|
|
|
;
|
|
; Ensure that the %ptrue from the preheader is sunk into the loop such that the ptest is removed.
|
|
;
|
|
|
|
define void @test_sink_ptrue_into_ptest(i32 %n) {
|
|
; CHECK-LABEL: test_sink_ptrue_into_ptest:
|
|
; CHECK: // %bb.0: // %entry
|
|
; CHECK-NEXT: whilelt p0.s, wzr, w0
|
|
; CHECK-NEXT: b.pl .LBB0_3
|
|
; CHECK-NEXT: // %bb.1: // %for.body.preheader
|
|
; CHECK-NEXT: mov w8, wzr
|
|
; CHECK-NEXT: cntw x9
|
|
; CHECK-NEXT: .LBB0_2: // %for.body
|
|
; CHECK-NEXT: // =>This Inner Loop Header: Depth=1
|
|
; CHECK-NEXT: whilelt p0.s, w8, w0
|
|
; CHECK-NEXT: add w8, w8, w9
|
|
; CHECK-NEXT: b.mi .LBB0_2
|
|
; CHECK-NEXT: .LBB0_3: // %exit
|
|
; CHECK-NEXT: ret
|
|
entry:
|
|
%vscale = tail call i32 @llvm.vscale.i32()
|
|
%step = shl nuw nsw i32 %vscale, 2
|
|
%ptrue.ph = tail call <vscale x 4 x i1> @llvm.aarch64.sve.ptrue.nxv4i1(i32 31)
|
|
%while.ph = tail call <vscale x 4 x i1> @llvm.aarch64.sve.whilelt.nxv4i1.i32(i32 0, i32 %n)
|
|
%ptest.ph = tail call i1 @llvm.aarch64.sve.ptest.first.nxv4i1(<vscale x 4 x i1> %ptrue.ph, <vscale x 4 x i1> %while.ph)
|
|
br i1 %ptest.ph, label %for.body, label %exit
|
|
|
|
for.body:
|
|
%i = phi i32 [ 0, %entry ], [ %i.next, %for.body ]
|
|
%i.next = add i32 %i, %step
|
|
%while = call <vscale x 4 x i1> @llvm.aarch64.sve.whilelt.nxv4i1.i32(i32 %i, i32 %n)
|
|
%ptest = call i1 @llvm.aarch64.sve.ptest.first.nxv4i1(<vscale x 4 x i1> %ptrue.ph, <vscale x 4 x i1> %while)
|
|
br i1 %ptest, label %for.body, label %exit
|
|
|
|
exit:
|
|
ret void
|
|
}
|
|
|
|
declare i32 @llvm.vscale.i32()
|
|
declare <vscale x 4 x i1> @llvm.aarch64.sve.ptrue.nxv4i1(i32 immarg)
|
|
declare <vscale x 4 x i1> @llvm.aarch64.sve.whilelt.nxv4i1.i32(i32, i32)
|
|
declare i1 @llvm.aarch64.sve.ptest.first.nxv4i1(<vscale x 4 x i1>, <vscale x 4 x i1>)
|