| ; 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 w9, wzr |
| ; CHECK-NEXT: cntw x8 |
| ; CHECK-NEXT: .LBB0_2: // %for.body |
| ; CHECK-NEXT: // =>This Inner Loop Header: Depth=1 |
| ; CHECK-NEXT: add w10, w9, w8 |
| ; CHECK-NEXT: whilelt p0.s, w9, w0 |
| ; CHECK-NEXT: mov w9, w10 |
| ; 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>) |