mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2024-12-12 22:17:47 +00:00
c3dd7f4b12
This patch adds the ability to peel off iterations of the first loop in loop
fusion. This can allow for both loops to have the same trip count, making it
legal for them to be fused together.
Here is a simple scenario peeling can be used in loop fusion:
for (i = 0; i < 10; ++i)
a[i] = a[i] + 3;
for (j = 1; j < 10; ++j)
b[j] = b[j] + 5;
Here is we can make use of peeling, and then fuse the two loops together. We
can peel off the 0th iteration of the loop i, and then combine loop i and j for
i = 1 to 10.
a[0] = a[0] +3;
for (i = 1; i < 10; ++i) {
a[i] = a[i] + 3;
b[i] = b[i] + 5;
}
Currently peeling with loop fusion is only supported for loops with constant
trip counts and a single exit point. Both unguarded and guarded loops are
supported.
Reviewed By: bmahjour (Bardia Mahjour), MaskRay (Fangrui Song)
Differential Revision: https://reviews.llvm.org/D82927
85 lines
3.4 KiB
LLVM
85 lines
3.4 KiB
LLVM
; RUN: opt -S -loop-fusion -loop-fusion-peel-max-count=3 < %s | FileCheck %s
|
|
|
|
; Tests that we do not fuse these two loops together. These loops do not have
|
|
; the same tripcount, and the first loop is valid candiate for peeling; however
|
|
; the loops are not adjacent, hence they are not valid to be fused (after
|
|
; peeling).
|
|
; The expected output of this test is the function below.
|
|
|
|
; CHECK-LABEL: void @function(i32* noalias %arg)
|
|
; CHECK-NEXT: for.first.preheader:
|
|
; CHECK-NEXT: br label %for.first
|
|
; CHECK: for.first:
|
|
; CHECK: br label %for.first.latch
|
|
; CHECK: for.first.latch:
|
|
; CHECK: br i1 %exitcond4, label %for.first, label %for.first.exit
|
|
; CHECK: for.first.exit:
|
|
; CHECK-NEXT: br label %for.next
|
|
; CHECK: for.next:
|
|
; CHECK-NEXT: br label %for.second.preheader
|
|
; CHECK: for.second.preheader:
|
|
; CHECK: br label %for.second
|
|
; CHECK: for.second:
|
|
; CHECK: br label %for.second.latch
|
|
; CHECK: for.second.latch:
|
|
; CHECK: br i1 %exitcond, label %for.second, label %for.end
|
|
; CHECK: for.end:
|
|
; CHECK-NEXT: ret void
|
|
|
|
@B = common global [1024 x i32] zeroinitializer, align 16
|
|
|
|
define void @function(i32* noalias %arg) {
|
|
for.first.preheader:
|
|
br label %for.first
|
|
|
|
for.first: ; preds = %for.first.preheader, %for.first.latch
|
|
%.014 = phi i32 [ 0, %for.first.preheader ], [ %tmp15, %for.first.latch ]
|
|
%indvars.iv23 = phi i64 [ 0, %for.first.preheader ], [ %indvars.iv.next3, %for.first.latch ]
|
|
%tmp = add nsw i32 %.014, -3
|
|
%tmp8 = add nuw nsw i64 %indvars.iv23, 3
|
|
%tmp9 = trunc i64 %tmp8 to i32
|
|
%tmp10 = mul nsw i32 %tmp, %tmp9
|
|
%tmp11 = trunc i64 %indvars.iv23 to i32
|
|
%tmp12 = srem i32 %tmp10, %tmp11
|
|
%tmp13 = getelementptr inbounds i32, i32* %arg, i64 %indvars.iv23
|
|
store i32 %tmp12, i32* %tmp13, align 4
|
|
br label %for.first.latch
|
|
|
|
for.first.latch: ; preds = %for.first
|
|
%indvars.iv.next3 = add nuw nsw i64 %indvars.iv23, 1
|
|
%tmp15 = add nuw nsw i32 %.014, 1
|
|
%exitcond4 = icmp ne i64 %indvars.iv.next3, 100
|
|
br i1 %exitcond4, label %for.first, label %for.first.exit
|
|
|
|
for.first.exit: ; preds: %for.first.latch
|
|
br label %for.next
|
|
|
|
for.next: ; preds = %for.first.exit
|
|
br label %for.second.preheader
|
|
|
|
for.second.preheader: ; preds = %for.next
|
|
br label %for.second
|
|
|
|
for.second: ; preds = %for.second.preheader, %for.second.latch
|
|
%.02 = phi i32 [ 0, %for.second.preheader ], [ %tmp28, %for.second.latch ]
|
|
%indvars.iv1 = phi i64 [ 3, %for.second.preheader ], [ %indvars.iv.next, %for.second.latch ]
|
|
%tmp20 = add nsw i32 %.02, -3
|
|
%tmp21 = add nuw nsw i64 %indvars.iv1, 3
|
|
%tmp22 = trunc i64 %tmp21 to i32
|
|
%tmp23 = mul nsw i32 %tmp20, %tmp22
|
|
%tmp24 = trunc i64 %indvars.iv1 to i32
|
|
%tmp25 = srem i32 %tmp23, %tmp24
|
|
%tmp26 = getelementptr inbounds [1024 x i32], [1024 x i32]* @B, i64 0, i64 %indvars.iv1
|
|
store i32 %tmp25, i32* %tmp26, align 4
|
|
br label %for.second.latch
|
|
|
|
for.second.latch: ; preds = %for.second
|
|
%indvars.iv.next = add nuw nsw i64 %indvars.iv1, 1
|
|
%tmp28 = add nuw nsw i32 %.02, 1
|
|
%exitcond = icmp ne i64 %indvars.iv.next, 100
|
|
br i1 %exitcond, label %for.second, label %for.end
|
|
|
|
for.end: ; preds = %for.second.latch
|
|
ret void
|
|
}
|