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