1; RUN: opt < %s -scalable-vectorization=on -force-target-supports-scalable-vectors=true -passes=loop-vectorize -force-vector-width=2 -force-vector-interleave=2 -S | FileCheck %s 2 3define void @test1(ptr noalias nocapture %a, ptr noalias nocapture readonly %b) { 4; CHECK-LABEL: @test1( 5; CHECK: vector.body: 6; CHECK: [[E1:%.*]] = extractelement <vscale x 2 x float> {{.+}}, i32 0 7; CHECK-NEXT: [[FCMP1:%.*]] = fcmp ogt float [[E1]] 8; CHECK-NEXT: [[E2:%.*]] = extractelement <vscale x 2 x float> {{.+}}, i32 0 9; CHECK-NEXT: [[FCMP2:%.*]] = fcmp ogt float [[E2]] 10; CHECK-NEXT: tail call void @llvm.assume(i1 [[FCMP1]]) 11; CHECK-NEXT: tail call void @llvm.assume(i1 [[FCMP2]]) 12entry: 13 br label %for.body 14 15for.body: ; preds = %for.body, %entry 16 %indvars.iv = phi i64 [ 0, %entry ], [ %indvars.iv.next, %for.body ] 17 %arrayidx = getelementptr inbounds float, ptr %b, i64 %indvars.iv 18 %0 = load float, ptr %arrayidx, align 4 19 %cmp1 = fcmp ogt float %0, 1.000000e+02 20 tail call void @llvm.assume(i1 %cmp1) 21 %add = fadd float %0, 1.000000e+00 22 %arrayidx5 = getelementptr inbounds float, ptr %a, i64 %indvars.iv 23 store float %add, ptr %arrayidx5, align 4 24 %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1 25 %exitcond = icmp eq i64 %indvars.iv, 1599 26 br i1 %exitcond, label %for.end, label %for.body, !llvm.loop !0 27 28for.end: ; preds = %for.body 29 ret void 30} 31 32declare void @llvm.assume(i1) #0 33 34attributes #0 = { nounwind willreturn } 35 36%struct.data = type { ptr, ptr } 37 38define void @test2(ptr %a, ptr %b) { 39; CHECK-LABEL: @test2( 40; CHECK: entry: 41; CHECK: [[MASKCOND:%.*]] = icmp eq i64 %ptrint1, 0 42; CHECK: [[MASKCOND4:%.*]] = icmp eq i64 %ptrint2, 0 43; CHECK: vector.body: 44; CHECK: tail call void @llvm.assume(i1 [[MASKCOND]]) 45; CHECK-NEXT: tail call void @llvm.assume(i1 [[MASKCOND]]) 46; CHECK: tail call void @llvm.assume(i1 [[MASKCOND4]]) 47; CHECK-NEXT: tail call void @llvm.assume(i1 [[MASKCOND4]]) 48entry: 49 %ptrint1 = ptrtoint ptr %a to i64 50 %maskcond = icmp eq i64 %ptrint1, 0 51 %ptrint2 = ptrtoint ptr %b to i64 52 %maskcond4 = icmp eq i64 %ptrint2, 0 53 br label %for.body 54 55 56for.body: ; preds = %for.body, %entry 57 %indvars.iv = phi i64 [ 0, %entry ], [ %indvars.iv.next, %for.body ] 58 tail call void @llvm.assume(i1 %maskcond) 59 %arrayidx = getelementptr inbounds float, ptr %a, i64 %indvars.iv 60 %0 = load float, ptr %arrayidx, align 4 61 %add = fadd float %0, 1.000000e+00 62 tail call void @llvm.assume(i1 %maskcond4) 63 %arrayidx5 = getelementptr inbounds float, ptr %b, i64 %indvars.iv 64 store float %add, ptr %arrayidx5, align 4 65 %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1 66 %exitcond = icmp eq i64 %indvars.iv, 1599 67 br i1 %exitcond, label %for.end, label %for.body, !llvm.loop !0 68 69for.end: ; preds = %for.body 70 ret void 71} 72 73; Test case for PR43620. Make sure we can vectorize with predication in presence 74; of assume calls. For now, check that we drop all assumes in predicated blocks 75; in the vector body. 76define void @predicated_assume(ptr noalias nocapture readonly %a, ptr noalias nocapture %b, i64 %n) { 77; Check that the vector.body does not contain any assumes. 78; CHECK-LABEL: @predicated_assume( 79; CHECK: vector.body: 80; CHECK-NOT: llvm.assume 81; CHECK: for.body: 82entry: 83 br label %for.body 84 85for.body: ; preds = %for.body.preheader, %if.end5 86 %indvars.iv = phi i64 [ 0, %entry ], [ %indvars.iv.next, %if.end5 ] 87 %cmp1 = icmp ult i64 %indvars.iv, 495616 88 br i1 %cmp1, label %if.end5, label %if.else 89 90if.else: ; preds = %for.body 91 %cmp2 = icmp ult i64 %indvars.iv, 991232 92 tail call void @llvm.assume(i1 %cmp2) 93 br label %if.end5 94 95if.end5: ; preds = %for.body, %if.else 96 %x.0 = phi float [ 4.200000e+01, %if.else ], [ 2.300000e+01, %for.body ] 97 %arrayidx = getelementptr inbounds float, ptr %a, i64 %indvars.iv 98 %0 = load float, ptr %arrayidx, align 4 99 %mul = fmul float %x.0, %0 100 %arrayidx7 = getelementptr inbounds float, ptr %b, i64 %indvars.iv 101 store float %mul, ptr %arrayidx7, align 4 102 %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1 103 %cmp = icmp eq i64 %indvars.iv.next, %n 104 br i1 %cmp, label %for.cond.cleanup, label %for.body, !llvm.loop !0 105 106for.cond.cleanup: ; preds = %if.end5, %entry 107 ret void 108} 109 110!0 = distinct !{!0, !1} 111!1 = !{!"llvm.loop.vectorize.scalable.enable", i1 true} 112