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
106 lines
4.0 KiB
LLVM
106 lines
4.0 KiB
LLVM
; RUN: opt -S -loop-fusion -loop-fusion-peel-max-count=3 < %s | FileCheck %s
|
|
|
|
; Tests whether we can fuse two loops together if they have constant but a
|
|
; different tripcount.
|
|
; The first three iterations of the first loop should be peeled, and then the
|
|
; two loops should be fused together in this example.
|
|
|
|
; C Code
|
|
;
|
|
; int B[1024];
|
|
;
|
|
; void function(int *arg) {
|
|
; for (int i = 0; i != 100; ++i)
|
|
; arg[i] = ((i - 3)*(i+3)) % i;
|
|
;
|
|
; for (int i = 3; i != 100; ++i)
|
|
; B[i] = ((i-6)*(i+3)) % i;
|
|
; }
|
|
|
|
; CHECK-LABEL: void @function(i32* noalias %arg)
|
|
; CHECK-NEXT: for.first.preheader:
|
|
; CHECK-NEXT: br label %for.first.peel.begin
|
|
; CHECK: for.first.peel.begin:
|
|
; CHECK-NEXT: br label %for.first.peel
|
|
; CHECK: for.first.peel:
|
|
; CHECK: br label %for.first.latch.peel
|
|
; CHECK: for.first.latch.peel:
|
|
; CHECK: br label %for.first.peel.next
|
|
; CHECK: for.first.peel.next:
|
|
; CHECK-NEXT: br label %for.first.peel2
|
|
; CHECK: for.first.peel2:
|
|
; CHECK: br label %for.first.latch.peel10
|
|
; CHECK: for.first.latch.peel10:
|
|
; CHECK: br label %for.first.peel.next1
|
|
; CHECK: for.first.peel.next1:
|
|
; CHECK-NEXT: br label %for.first.peel15
|
|
; CHECK: for.first.peel15:
|
|
; CHECK: br label %for.first.latch.peel23
|
|
; CHECK: for.first.latch.peel23:
|
|
; CHECK: br label %for.first.peel.next14
|
|
; CHECK: for.first.peel.next14:
|
|
; CHECK-NEXT: br label %for.first.peel.next27
|
|
; CHECK: for.first.peel.next27:
|
|
; CHECK-NEXT: br label %for.first.preheader.peel.newph
|
|
; CHECK: for.first.preheader.peel.newph:
|
|
; CHECK-NEXT: br label %for.first
|
|
; CHECK: for.first:
|
|
; CHECK: br label %for.first.latch
|
|
; CHECK: for.first.latch:
|
|
; CHECK: br label %for.second.latch
|
|
; CHECK: for.second.latch:
|
|
; CHECK: br i1 %exitcond, label %for.first, 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.second.preheader
|
|
|
|
for.second.preheader: ; preds = %for.first.latch
|
|
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
|
|
}
|