| ; RUN: opt -mtriple=thumbv8.1m.main -mattr=+mve.fp -passes=loop-unroll -S < %s -o - | FileCheck %s |
| |
| ; CHECK-LABEL: @loopfn |
| ; CHECK: vector.body: |
| ; CHECK: br i1 %7, label %middle.block, label %vector.body, !llvm.loop !0 |
| ; CHECK: middle.block: |
| ; CHECK: br i1 %cmp.n, label %for.cond.cleanup, label %for.body.preheader13 |
| ; CHECK: for.body: |
| ; CHECK: br i1 %exitcond, label %for.cond.cleanup.loopexit, label %for.body, !llvm.loop !2 |
| |
| define void @loopfn(ptr %s1, ptr %s2, ptr %d, i32 %n) { |
| entry: |
| %cmp10 = icmp sgt i32 %n, 0 |
| br i1 %cmp10, label %for.body.preheader, label %for.cond.cleanup |
| |
| for.body.preheader: ; preds = %entry |
| %min.iters.check = icmp ult i32 %n, 4 |
| br i1 %min.iters.check, label %for.body.preheader13, label %vector.ph |
| |
| for.body.preheader13: ; preds = %middle.block, %for.body.preheader |
| %i.011.ph = phi i32 [ 0, %for.body.preheader ], [ %n.vec, %middle.block ] |
| br label %for.body |
| |
| vector.ph: ; preds = %for.body.preheader |
| %n.vec = and i32 %n, -4 |
| br label %vector.body |
| |
| vector.body: ; preds = %vector.body, %vector.ph |
| %index = phi i32 [ 0, %vector.ph ], [ %index.next, %vector.body ] |
| %0 = getelementptr inbounds float, ptr %s1, i32 %index |
| %1 = bitcast ptr %0 to ptr |
| %wide.load = load <4 x float>, ptr %1, align 4 |
| %2 = getelementptr inbounds float, ptr %s2, i32 %index |
| %3 = bitcast ptr %2 to ptr |
| %wide.load12 = load <4 x float>, ptr %3, align 4 |
| %4 = fadd fast <4 x float> %wide.load12, %wide.load |
| %5 = getelementptr inbounds float, ptr %d, i32 %index |
| %6 = bitcast ptr %5 to ptr |
| store <4 x float> %4, ptr %6, align 4 |
| %index.next = add i32 %index, 4 |
| %7 = icmp eq i32 %index.next, %n.vec |
| br i1 %7, label %middle.block, label %vector.body, !llvm.loop !0 |
| |
| middle.block: ; preds = %vector.body |
| %cmp.n = icmp eq i32 %n.vec, %n |
| br i1 %cmp.n, label %for.cond.cleanup, label %for.body.preheader13 |
| |
| for.cond.cleanup.loopexit: ; preds = %for.body |
| br label %for.cond.cleanup |
| |
| for.cond.cleanup: ; preds = %for.cond.cleanup.loopexit, %middle.block, %entry |
| ret void |
| |
| for.body: ; preds = %for.body.preheader13, %for.body |
| %i.011 = phi i32 [ %add3, %for.body ], [ %i.011.ph, %for.body.preheader13 ] |
| %arrayidx = getelementptr inbounds float, ptr %s1, i32 %i.011 |
| %8 = load float, ptr %arrayidx, align 4 |
| %arrayidx1 = getelementptr inbounds float, ptr %s2, i32 %i.011 |
| %9 = load float, ptr %arrayidx1, align 4 |
| %add = fadd fast float %9, %8 |
| %arrayidx2 = getelementptr inbounds float, ptr %d, i32 %i.011 |
| store float %add, ptr %arrayidx2, align 4 |
| %add3 = add nuw nsw i32 %i.011, 1 |
| %exitcond = icmp eq i32 %add3, %n |
| br i1 %exitcond, label %for.cond.cleanup.loopexit, label %for.body, !llvm.loop !2 |
| } |
| |
| |
| ; Same as above but without the nounroll on the remainder loop. Neither loop should be unrolled. |
| |
| ; CHECK-LABEL: @remainder |
| ; CHECK: vector.body: |
| ; CHECK: br i1 %7, label %middle.block, label %vector.body, !llvm.loop !0 |
| ; CHECK: middle.block: |
| ; CHECK: br i1 %cmp.n, label %for.cond.cleanup, label %for.body.preheader13 |
| ; CHECK: for.body: |
| ; CHECK: br i1 %exitcond, label %for.cond.cleanup.loopexit, label %for.body, !llvm.loop !0 |
| |
| define void @remainder(ptr %s1, ptr %s2, ptr %d, i32 %n) { |
| entry: |
| %cmp10 = icmp sgt i32 %n, 0 |
| br i1 %cmp10, label %for.body.preheader, label %for.cond.cleanup |
| |
| for.body.preheader: ; preds = %entry |
| %min.iters.check = icmp ult i32 %n, 4 |
| br i1 %min.iters.check, label %for.body.preheader13, label %vector.ph |
| |
| for.body.preheader13: ; preds = %middle.block, %for.body.preheader |
| %i.011.ph = phi i32 [ 0, %for.body.preheader ], [ %n.vec, %middle.block ] |
| br label %for.body |
| |
| vector.ph: ; preds = %for.body.preheader |
| %n.vec = and i32 %n, -4 |
| br label %vector.body |
| |
| vector.body: ; preds = %vector.body, %vector.ph |
| %index = phi i32 [ 0, %vector.ph ], [ %index.next, %vector.body ] |
| %0 = getelementptr inbounds float, ptr %s1, i32 %index |
| %1 = bitcast ptr %0 to ptr |
| %wide.load = load <4 x float>, ptr %1, align 4 |
| %2 = getelementptr inbounds float, ptr %s2, i32 %index |
| %3 = bitcast ptr %2 to ptr |
| %wide.load12 = load <4 x float>, ptr %3, align 4 |
| %4 = fadd fast <4 x float> %wide.load12, %wide.load |
| %5 = getelementptr inbounds float, ptr %d, i32 %index |
| %6 = bitcast ptr %5 to ptr |
| store <4 x float> %4, ptr %6, align 4 |
| %index.next = add i32 %index, 4 |
| %7 = icmp eq i32 %index.next, %n.vec |
| br i1 %7, label %middle.block, label %vector.body, !llvm.loop !0 |
| |
| middle.block: ; preds = %vector.body |
| %cmp.n = icmp eq i32 %n.vec, %n |
| br i1 %cmp.n, label %for.cond.cleanup, label %for.body.preheader13 |
| |
| for.cond.cleanup.loopexit: ; preds = %for.body |
| br label %for.cond.cleanup |
| |
| for.cond.cleanup: ; preds = %for.cond.cleanup.loopexit, %middle.block, %entry |
| ret void |
| |
| for.body: ; preds = %for.body.preheader13, %for.body |
| %i.011 = phi i32 [ %add3, %for.body ], [ %i.011.ph, %for.body.preheader13 ] |
| %arrayidx = getelementptr inbounds float, ptr %s1, i32 %i.011 |
| %8 = load float, ptr %arrayidx, align 4 |
| %arrayidx1 = getelementptr inbounds float, ptr %s2, i32 %i.011 |
| %9 = load float, ptr %arrayidx1, align 4 |
| %add = fadd fast float %9, %8 |
| %arrayidx2 = getelementptr inbounds float, ptr %d, i32 %i.011 |
| store float %add, ptr %arrayidx2, align 4 |
| %add3 = add nuw nsw i32 %i.011, 1 |
| %exitcond = icmp eq i32 %add3, %n |
| br i1 %exitcond, label %for.cond.cleanup.loopexit, label %for.body, !llvm.loop !0 |
| } |
| |
| |
| |
| ; CHECK-LABEL: @nested |
| ; CHECK: for.outer: |
| ; CHECK: br label %vector.body |
| ; CHECK: vector.body: |
| ; CHECK: br i1 %8, label %for.latch, label %vector.body, !llvm.loop !0 |
| ; CHECK: for.latch: |
| ; CHECK: br i1 %exitcond34, label %for.cond.cleanup.loopexit, label %for.outer |
| |
| define void @nested(ptr %s1, ptr %s2, ptr %d, i32 %n) { |
| entry: |
| %cmp31 = icmp eq i32 %n, 0 |
| br i1 %cmp31, label %for.cond.cleanup, label %for.outer.preheader |
| |
| for.outer.preheader: ; preds = %entry |
| %min.iters.check = icmp ult i32 %n, 4 |
| %n.vec = and i32 %n, -4 |
| %cmp.n = icmp eq i32 %n.vec, %n |
| br label %for.outer |
| |
| for.outer: ; preds = %for.outer.preheader, %for.cond1.for.cond.cleanup3_crit_edge.us |
| %j.032.us = phi i32 [ %inc.us, %for.latch ], [ 0, %for.outer.preheader ] |
| %mul.us = mul i32 %j.032.us, %n |
| br label %vector.body |
| |
| vector.body: ; preds = %for.outer, %vector.body |
| %index = phi i32 [ %index.next, %vector.body ], [ 0, %for.outer ] |
| %0 = add i32 %index, %mul.us |
| %1 = getelementptr inbounds float, ptr %s1, i32 %0 |
| %2 = bitcast ptr %1 to ptr |
| %wide.load = load <4 x float>, ptr %2, align 4 |
| %3 = getelementptr inbounds float, ptr %s2, i32 %0 |
| %4 = bitcast ptr %3 to ptr |
| %wide.load35 = load <4 x float>, ptr %4, align 4 |
| %5 = fadd fast <4 x float> %wide.load35, %wide.load |
| %6 = getelementptr inbounds float, ptr %d, i32 %0 |
| %7 = bitcast ptr %6 to ptr |
| store <4 x float> %5, ptr %7, align 4 |
| %index.next = add i32 %index, 4 |
| %8 = icmp eq i32 %index.next, %n.vec |
| br i1 %8, label %for.latch, label %vector.body, !llvm.loop !0 |
| |
| for.latch: ; preds = %vector.body, %for.outer |
| %i.030.us.ph = phi i32 [ %n.vec, %vector.body ] |
| %inc.us = add nuw i32 %j.032.us, 1 |
| %exitcond34 = icmp eq i32 %inc.us, %n |
| br i1 %exitcond34, label %for.cond.cleanup.loopexit, label %for.outer |
| |
| for.cond.cleanup.loopexit: |
| br label %for.cond.cleanup |
| |
| for.cond.cleanup: ; preds = %for.cond1.for.cond.cleanup3_crit_edge.us, %entry |
| ret void |
| } |
| |
| ; Test that we don't unroll loops that only contain vector intrinsics. |
| ; CHECK-LABEL: test_intrinsics |
| ; CHECK: call <16 x i8> @llvm.arm.mve.sub |
| ; CHECK-NOT: call <16 x i8> @llvm.arm.mve.sub |
| define dso_local arm_aapcs_vfpcc void @test_intrinsics(ptr noalias nocapture readonly %a, ptr noalias nocapture readonly %b, ptr noalias nocapture %c, i32 %N) { |
| entry: |
| %cmp8 = icmp eq i32 %N, 0 |
| %tmp8 = add i32 %N, 15 |
| %tmp9 = lshr i32 %tmp8, 4 |
| %tmp10 = shl nuw i32 %tmp9, 4 |
| %tmp11 = add i32 %tmp10, -16 |
| %tmp12 = lshr i32 %tmp11, 4 |
| %tmp13 = add nuw nsw i32 %tmp12, 1 |
| br i1 %cmp8, label %for.cond.cleanup, label %vector.ph |
| |
| vector.ph: |
| br label %vector.body |
| |
| vector.body: |
| %index = phi i32 [ 0, %vector.ph ], [ %index.next, %vector.body ] |
| %tmp14 = phi i32 [ %tmp13, %vector.ph ], [ %tmp15, %vector.body ] |
| %0 = phi i32 [ %N, %vector.ph ], [ %2, %vector.body ] |
| %tmp = getelementptr inbounds i8, ptr %a, i32 %index |
| %1 = call <16 x i1> @llvm.arm.mve.vctp8(i32 %0) |
| %2 = sub i32 %0, 16 |
| %tmp2 = bitcast ptr %tmp to ptr |
| %wide.masked.load = tail call <16 x i8> @llvm.masked.load.v16i8.p0(ptr %tmp2, i32 4, <16 x i1> %1, <16 x i8> undef) |
| %tmp3 = getelementptr inbounds i8, ptr %b, i32 %index |
| %tmp4 = bitcast ptr %tmp3 to ptr |
| %wide.masked.load2 = tail call <16 x i8> @llvm.masked.load.v16i8.p0(ptr %tmp4, i32 4, <16 x i1> %1, <16 x i8> undef) |
| %sub = call <16 x i8> @llvm.arm.mve.sub.predicated.v16i8.v16i1(<16 x i8> %wide.masked.load2, <16 x i8> %wide.masked.load, <16 x i1> %1, <16 x i8> undef) |
| %tmp6 = getelementptr inbounds i8, ptr %c, i32 %index |
| %tmp7 = bitcast ptr %tmp6 to ptr |
| tail call void @llvm.masked.store.v16i8.p0(<16 x i8> %sub, ptr %tmp7, i32 4, <16 x i1> %1) |
| %index.next = add i32 %index, 16 |
| %tmp15 = sub i32 %tmp14, 1 |
| %tmp16 = icmp ne i32 %tmp15, 0 |
| br i1 %tmp16, label %vector.body, label %for.cond.cleanup |
| |
| for.cond.cleanup: ; preds = %vector.body, %entry |
| ret void |
| } |
| |
| declare <16 x i1> @llvm.arm.mve.vctp8(i32) |
| declare <16 x i8> @llvm.masked.load.v16i8.p0(ptr, i32, <16 x i1>, <16 x i8>) |
| declare <16 x i8> @llvm.arm.mve.sub.predicated.v16i8.v16i1(<16 x i8>, <16 x i8>, <16 x i1>, <16 x i8>) |
| declare void @llvm.masked.store.v16i8.p0(<16 x i8>, ptr, i32, <16 x i1>) |
| |
| |
| !0 = distinct !{!0, !1} |
| !1 = !{!"llvm.loop.isvectorized", i32 1} |
| !2 = distinct !{!2, !3, !1} |
| !3 = !{!"llvm.loop.unroll.runtime.disable"} |
