183 lines · plain
1; RUN: opt -passes=loop-idiom -use-lir-code-size-heurs=true < %s -S | FileCheck %s2 3; When compiling for codesize we avoid idiom recognition for a4; multi-block loop unless it is one of5; - a loop_memset idiom, or6; - a memset/memcpy idiom in a nested loop.7 8declare void @llvm.memset.p0.i64(ptr nocapture, i8, i64, i1)9@APPLES = common global i32 0, align 410@ORANGES = common global i32 0, align 411 12; LIR allowed: loop_memset idiom in multi-block loop.13; ===================================================14; CHECK-LABEL: @LoopMemset15; CHECK: for.body.preheader:16; CHECK: call void @llvm.memset17; CHECK: for.body:18;19define i32 @LoopMemset(ptr noalias nocapture %DST, i32 %SIZE) local_unnamed_addr optsize {20entry:21 %cmp12 = icmp sgt i32 %SIZE, 022 br i1 %cmp12, label %for.body.preheader, label %for.end23 24for.body.preheader: ; preds = %entry25 br label %for.body26 27for.body: ; preds = %for.body.preheader, %for.inc28 %indvars.iv = phi i64 [ 0, %for.body.preheader ], [ %indvars.iv.next, %for.inc ]29 %BASKET.013 = phi i32 [ %BASKET.1, %for.inc ], [ 0, %for.body.preheader ]30 %arraydecay = getelementptr inbounds [2048 x i8], ptr %DST, i64 %indvars.iv, i64 031 tail call void @llvm.memset.p0.i64(ptr %arraydecay, i8 -1, i64 2048, i1 false)32 %0 = trunc i64 %indvars.iv to i3233 %rem11 = and i32 %0, 134 %cmp1 = icmp eq i32 %rem11, 035 %1 = load i32, ptr @ORANGES, align 436 %2 = load i32, ptr @APPLES, align 437 br i1 %cmp1, label %if.then, label %if.else38 39if.else: ; preds = %for.body40 %dec3 = add nsw i32 %2, -141 store i32 %dec3, ptr @APPLES, align 442 br label %for.inc43 44if.then: ; preds = %for.body45 %dec = add nsw i32 %1, -146 store i32 %dec, ptr @ORANGES, align 447 br label %for.inc48 49for.inc: ; preds = %if.then, %if.else50 %.pn = phi i32 [ %2, %if.then ], [ %1, %if.else ]51 %BASKET.1 = add nsw i32 %.pn, %BASKET.01352 %indvars.iv.next = add nuw nsw i64 %indvars.iv, 153 %lftr.wideiv = trunc i64 %indvars.iv.next to i3254 %exitcond = icmp ne i32 %lftr.wideiv, %SIZE55 br i1 %exitcond, label %for.body, label %for.end.loopexit56 57for.end.loopexit: ; preds = %for.inc58 %BASKET.1.lcssa = phi i32 [ %BASKET.1, %for.inc ]59 br label %for.end60 61for.end: ; preds = %for.end.loopexit, %entry62 %BASKET.0.lcssa = phi i32 [ 0, %entry ], [ %BASKET.1.lcssa, %for.end.loopexit ]63 ret i32 %BASKET.0.lcssa64}65 66; LIR allowed: memset idiom in multi-block nested loop,67; which is recognized as a loop_memset in its turn.68; =====================================================69; CHECK-LABEL: @NestedMemset_LoopMemset70; CHECK: for.cond1.preheader.preheader:71; CHECK: call void @llvm.memset72; CHECK: for.cond1.preheader:73;74define i32 @NestedMemset_LoopMemset(ptr noalias nocapture %DST, i32 %SIZE) local_unnamed_addr optsize {75entry:76 %cmp25 = icmp sgt i32 %SIZE, 077 br i1 %cmp25, label %for.cond1.preheader.preheader, label %for.end1178 79for.cond1.preheader.preheader: ; preds = %entry80 br label %for.cond1.preheader81 82for.cond1.preheader: ; preds = %for.cond1.preheader.preheader, %for.inc983 %i.027 = phi i32 [ %inc10, %for.inc9 ], [ 0, %for.cond1.preheader.preheader ]84 %BASKET.026 = phi i32 [ %BASKET.2.lcssa, %for.inc9 ], [ 0, %for.cond1.preheader.preheader ]85 %idxprom4 = sext i32 %i.027 to i6486 %rem22 = and i32 %i.027, 187 %cmp6 = icmp eq i32 %rem22, 088 br label %for.body389 90for.body3: ; preds = %for.cond1.preheader, %for.inc91 %indvars.iv = phi i64 [ 0, %for.cond1.preheader ], [ %indvars.iv.next, %for.inc ]92 %BASKET.123 = phi i32 [ %BASKET.026, %for.cond1.preheader ], [ %BASKET.2, %for.inc ]93 %arrayidx5 = getelementptr inbounds [2046 x i8], ptr %DST, i64 %idxprom4, i64 %indvars.iv94 store i8 -1, ptr %arrayidx5, align 195 %0 = load i32, ptr @APPLES, align 496 %1 = load i32, ptr @ORANGES, align 497 br i1 %cmp6, label %if.then, label %if.else98 99if.else: ; preds = %for.body3100 %dec8 = add nsw i32 %0, -1101 store i32 %dec8, ptr @APPLES, align 4102 br label %for.inc103 104if.then: ; preds = %for.body3105 %dec = add nsw i32 %1, -1106 store i32 %dec, ptr @ORANGES, align 4107 br label %for.inc108 109for.inc: ; preds = %if.then, %if.else110 %.pn = phi i32 [ %0, %if.then ], [ %1, %if.else ]111 %BASKET.2 = add nsw i32 %.pn, %BASKET.123112 %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1113 %exitcond = icmp ne i64 %indvars.iv.next, 2046114 br i1 %exitcond, label %for.body3, label %for.inc9115 116for.inc9: ; preds = %for.inc117 %BASKET.2.lcssa = phi i32 [ %BASKET.2, %for.inc ]118 %inc10 = add nsw i32 %i.027, 1119 %cmp = icmp slt i32 %inc10, %SIZE120 br i1 %cmp, label %for.cond1.preheader, label %for.end11.loopexit121 122for.end11.loopexit: ; preds = %for.inc9123 %BASKET.2.lcssa.lcssa = phi i32 [ %BASKET.2.lcssa, %for.inc9 ]124 br label %for.end11125 126for.end11: ; preds = %for.end11.loopexit, %entry127 %BASKET.0.lcssa = phi i32 [ 0, %entry ], [ %BASKET.2.lcssa.lcssa, %for.end11.loopexit ]128 ret i32 %BASKET.0.lcssa129}130 131; LIR avoided: memset idiom in multi-block top-level loop.132; ========================================================133; CHECK-LABEL: @Non_NestedMemset134; CHECK-NOT: call void @llvm.memset135;136define i32 @Non_NestedMemset(ptr noalias nocapture %DST, i32 %SIZE) local_unnamed_addr optsize {137entry:138 %cmp12 = icmp sgt i32 %SIZE, 0139 br i1 %cmp12, label %for.body.preheader, label %for.end140 141for.body.preheader: ; preds = %entry142 br label %for.body143 144for.body: ; preds = %for.body.preheader, %for.inc145 %indvars.iv = phi i64 [ 0, %for.body.preheader ], [ %indvars.iv.next, %for.inc ]146 %BASKET.013 = phi i32 [ %BASKET.1, %for.inc ], [ 0, %for.body.preheader ]147 %arrayidx = getelementptr inbounds i8, ptr %DST, i64 %indvars.iv148 store i8 -1, ptr %arrayidx, align 1149 %0 = trunc i64 %indvars.iv to i32150 %rem11 = and i32 %0, 1151 %cmp1 = icmp eq i32 %rem11, 0152 %1 = load i32, ptr @ORANGES, align 4153 %2 = load i32, ptr @APPLES, align 4154 br i1 %cmp1, label %if.then, label %if.else155 156if.else: ; preds = %for.body157 %dec3 = add nsw i32 %2, -1158 store i32 %dec3, ptr @APPLES, align 4159 br label %for.inc160 161if.then: ; preds = %for.body162 %dec = add nsw i32 %1, -1163 store i32 %dec, ptr @ORANGES, align 4164 br label %for.inc165 166for.inc: ; preds = %if.then, %if.else167 %.pn = phi i32 [ %2, %if.then ], [ %1, %if.else ]168 %BASKET.1 = add nsw i32 %.pn, %BASKET.013169 %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1170 %lftr.wideiv = trunc i64 %indvars.iv.next to i32171 %exitcond = icmp ne i32 %lftr.wideiv, %SIZE172 br i1 %exitcond, label %for.body, label %for.end.loopexit173 174for.end.loopexit: ; preds = %for.inc175 %BASKET.1.lcssa = phi i32 [ %BASKET.1, %for.inc ]176 br label %for.end177 178for.end: ; preds = %for.end.loopexit, %entry179 %BASKET.0.lcssa = phi i32 [ 0, %entry ], [ %BASKET.1.lcssa, %for.end.loopexit ]180 ret i32 %BASKET.0.lcssa181}182 183