71 changes: 71 additions & 0 deletions llvm/test/Transforms/CanonicalizeFreezeInLoops/func_from_mcf_r.ll
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,71 @@
; RUN: opt < %s -canon-freeze -S | FileCheck %s
; REQUIRES: aarch64-registered-target
target datalayout = "e-m:e-i8:8:32-i16:16:32-i64:64-i128:128-n32:64-S128"
target triple = "aarch64-unknown-linux-gnu"

%struct.arc = type { i32 }
%struct.g = type { i64, %struct.arc, i64, i64, i64 }

@m = global i64 0
@h = global %struct.arc* null
@j = global %struct.g zeroinitializer

define dso_local i32 @main() {
bb:
%tmp = load i64, i64* getelementptr inbounds (%struct.g, %struct.g* @j, i32 0, i32 0), align 8
%tmp1 = icmp sgt i64 %tmp, 0
br i1 %tmp1, label %bb2, label %bb35

bb2: ; preds = %bb
%tmp3 = load i64, i64* @m, align 8
%tmp4 = load %struct.arc*, %struct.arc** @h, align 8
; CHECK: %tmp3.frozen = freeze i64 %tmp3
br label %bb5

bb5: ; preds = %bb28, %bb2
%tmp6 = phi %struct.arc* [ %tmp4, %bb2 ], [ %tmp31, %bb28 ]
%tmp7 = phi i64 [ %tmp3, %bb2 ], [ %tmp12, %bb28 ]
; CHECK: %tmp7 = phi i64 [ %tmp3.frozen, %bb2 ], [ %tmp12, %bb28 ]
%tmp8 = phi i64 [ 0, %bb2 ], [ %tmp11, %bb28 ]
%tmp9 = trunc i64 %tmp7 to i32
%tmp10 = getelementptr inbounds %struct.arc, %struct.arc* %tmp6, i64 0, i32 0
store i32 %tmp9, i32* %tmp10, align 4
%tmp11 = add nuw nsw i64 %tmp8, 1
%tmp12 = add nsw i64 %tmp7, 1
; CHECK: %tmp12 = add i64 %tmp7, 1
store i64 %tmp12, i64* @m, align 8
%tmp13 = load i64, i64* inttoptr (i64 16 to i64*), align 16
%tmp14 = freeze i64 %tmp12
; CHECK-NOT: %tmp14 = freeze i64 %tmp12
%tmp15 = freeze i64 %tmp13
%tmp16 = sdiv i64 %tmp14, %tmp15
%tmp17 = mul i64 %tmp16, %tmp15
%tmp18 = sub i64 %tmp14, %tmp17
%tmp19 = load i64, i64* inttoptr (i64 24 to i64*), align 8
%tmp20 = icmp sgt i64 %tmp18, %tmp19
%tmp21 = load i64, i64* inttoptr (i64 32 to i64*), align 32
br i1 %tmp20, label %bb22, label %bb28

bb22: ; preds = %bb5
%tmp23 = mul nsw i64 %tmp21, %tmp19
%tmp24 = sub nsw i64 %tmp18, %tmp19
%tmp25 = add nsw i64 %tmp21, -1
%tmp26 = mul nsw i64 %tmp25, %tmp24
%tmp27 = add nsw i64 %tmp26, %tmp23
br label %bb28

bb28: ; preds = %bb22, %bb5
%tmp29 = phi i64 [ %tmp27, %bb22 ], [ %tmp21, %bb5 ]
%tmp30 = add nsw i64 %tmp29, %tmp16
%tmp31 = getelementptr inbounds %struct.arc, %struct.arc* getelementptr inbounds (%struct.g, %struct.g* @j, i32 0, i32 1), i64 %tmp30
store %struct.arc* %tmp31, %struct.arc** @h, align 8
%tmp32 = load i64, i64* getelementptr inbounds (%struct.g, %struct.g* @j, i32 0, i32 0), align 8
%tmp33 = icmp slt i64 %tmp11, %tmp32
br i1 %tmp33, label %bb5, label %bb34

bb34: ; preds = %bb28
br label %bb35

bb35: ; preds = %bb34, %bb
ret i32 0
}
34 changes: 34 additions & 0 deletions llvm/test/Transforms/CanonicalizeFreezeInLoops/nonsteps-preserve-flags.ll
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,34 @@
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
; RUN: opt < %s -canon-freeze -S | FileCheck %s
declare void @call(i32)

define void @add(i32 %init, i32 %n) {
; CHECK-LABEL: @add(
; CHECK-NEXT: entry:
; CHECK-NEXT: [[INIT_FROZEN:%.*]] = freeze i32 [[INIT:%.*]]
; CHECK-NEXT: br label [[LOOP:%.*]]
; CHECK: loop:
; CHECK-NEXT: [[I:%.*]] = phi i32 [ [[INIT_FROZEN]], [[ENTRY:%.*]] ], [ [[I_NEXT:%.*]], [[LOOP]] ]
; CHECK-NEXT: [[I_NEXT]] = add i32 [[I]], 1
; CHECK-NEXT: [[NONSTEP:%.*]] = mul nsw i32 [[I]], 2
; CHECK-NEXT: call void @call(i32 [[NONSTEP]])
; CHECK-NEXT: [[COND:%.*]] = icmp eq i32 [[I_NEXT]], [[N:%.*]]
; CHECK-NEXT: br i1 [[COND]], label [[LOOP]], label [[EXIT:%.*]]
; CHECK: exit:
; CHECK-NEXT: ret void
;
entry:
br label %loop

loop:
%i = phi i32 [ %init, %entry], [%i.next, %loop ]
%i.next = add nsw i32 %i, 1
%i.next.fr = freeze i32 %i.next
%nonstep = mul nsw i32 %i, 2
call void @call(i32 %nonstep)
%cond = icmp eq i32 %i.next.fr, %n
br i1 %cond, label %loop, label %exit

exit:
ret void
}
547 changes: 547 additions & 0 deletions llvm/test/Transforms/CanonicalizeFreezeInLoops/onephi.ll
View file

Large diffs are not rendered by default.

114 changes: 114 additions & 0 deletions llvm/test/Transforms/CanonicalizeFreezeInLoops/phis.ll
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,114 @@
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
; RUN: opt < %s -canon-freeze -S | FileCheck %s
; A set of tests that have several phi nodes
declare void @call(i32)
declare i32 @call2()

define void @onephi_used(i32 %n, i32 %i.init, i32 %j.init) {
; CHECK-LABEL: @onephi_used(
; CHECK-NEXT: entry:
; CHECK-NEXT: [[I_INIT_FROZEN:%.*]] = freeze i32 [[I_INIT:%.*]]
; CHECK-NEXT: br label [[LOOP:%.*]]
; CHECK: loop:
; CHECK-NEXT: [[I:%.*]] = phi i32 [ [[I_INIT_FROZEN]], [[ENTRY:%.*]] ], [ [[I_NEXT:%.*]], [[LOOP]] ]
; CHECK-NEXT: [[J:%.*]] = phi i32 [ [[J_INIT:%.*]], [[ENTRY]] ], [ [[J_NEXT:%.*]], [[LOOP]] ]
; CHECK-NEXT: [[I_NEXT]] = add i32 [[I]], 1
; CHECK-NEXT: [[J_NEXT]] = add nuw nsw i32 [[J]], -2
; CHECK-NEXT: call void @call(i32 [[I]])
; CHECK-NEXT: [[COND:%.*]] = icmp eq i32 [[I_NEXT]], [[N:%.*]]
; CHECK-NEXT: br i1 [[COND]], label [[LOOP]], label [[EXIT:%.*]]
; CHECK: exit:
; CHECK-NEXT: ret void
;
entry:
br label %loop
loop:
%i = phi i32 [ %i.init, %entry ], [ %i.next, %loop ]
%j = phi i32 [ %j.init, %entry ], [ %j.next, %loop ]
%i.next = add nuw nsw i32 %i, 1
%j.next = add nuw nsw i32 %j, -2
%i.fr = freeze i32 %i
call void @call(i32 %i.fr)
%cond = icmp eq i32 %i.next, %n
br i1 %cond, label %loop, label %exit

exit:
ret void
}

; Negative test
define void @twophis_used(i32 %n, i32 %i.init, i32 %j.init) {
; CHECK-LABEL: @twophis_used(
; CHECK-NEXT: entry:
; CHECK-NEXT: br label [[LOOP:%.*]]
; CHECK: loop:
; CHECK-NEXT: [[I:%.*]] = phi i32 [ [[I_INIT:%.*]], [[ENTRY:%.*]] ], [ [[I_NEXT:%.*]], [[LOOP]] ]
; CHECK-NEXT: [[J:%.*]] = phi i32 [ [[J_INIT:%.*]], [[ENTRY]] ], [ [[J_NEXT:%.*]], [[LOOP]] ]
; CHECK-NEXT: [[I_NEXT]] = add nuw nsw i32 [[I]], 1
; CHECK-NEXT: [[J_NEXT]] = add nuw nsw i32 [[J]], -2
; CHECK-NEXT: [[IJ:%.*]] = add i32 [[I]], [[J]]
; CHECK-NEXT: [[IJ_FR:%.*]] = freeze i32 [[IJ]]
; CHECK-NEXT: call void @call(i32 [[IJ_FR]])
; CHECK-NEXT: [[COND:%.*]] = icmp eq i32 [[I_NEXT]], [[N:%.*]]
; CHECK-NEXT: br i1 [[COND]], label [[LOOP]], label [[EXIT:%.*]]
; CHECK: exit:
; CHECK-NEXT: ret void
;
entry:
br label %loop
loop:
%i = phi i32 [ %i.init, %entry ], [ %i.next, %loop ]
%j = phi i32 [ %j.init, %entry ], [ %j.next, %loop ]
%i.next = add nuw nsw i32 %i, 1
%j.next = add nuw nsw i32 %j, -2
%ij = add i32 %i, %j
%ij.fr = freeze i32 %ij
call void @call(i32 %ij.fr)
%cond = icmp eq i32 %i.next, %n
br i1 %cond, label %loop, label %exit

exit:
ret void
}

; Negative test
define void @nonindphi_used(i32 %n, i32 %i.init, i32 %j.init, i32 %k.init) {
; CHECK-LABEL: @nonindphi_used(
; CHECK-NEXT: entry:
; CHECK-NEXT: br label [[LOOP:%.*]]
; CHECK: loop:
; CHECK-NEXT: [[I:%.*]] = phi i32 [ [[I_INIT:%.*]], [[ENTRY:%.*]] ], [ [[I_NEXT:%.*]], [[LOOP]] ]
; CHECK-NEXT: [[J:%.*]] = phi i32 [ [[J_INIT:%.*]], [[ENTRY]] ], [ [[J_NEXT:%.*]], [[LOOP]] ]
; CHECK-NEXT: [[K:%.*]] = phi i32 [ [[K_INIT:%.*]], [[ENTRY]] ], [ [[ANY:%.*]], [[LOOP]] ]
; CHECK-NEXT: [[I_NEXT]] = add nuw nsw i32 [[I]], 1
; CHECK-NEXT: [[J_NEXT]] = add nuw nsw i32 [[J]], -2
; CHECK-NEXT: [[IJ:%.*]] = add i32 [[I]], [[J]]
; CHECK-NEXT: [[IJK:%.*]] = add i32 [[IJ]], [[K]]
; CHECK-NEXT: [[IJK_FR:%.*]] = freeze i32 [[IJK]]
; CHECK-NEXT: call void @call(i32 [[IJK_FR]])
; CHECK-NEXT: [[ANY]] = call i32 @call2()
; CHECK-NEXT: [[COND:%.*]] = icmp eq i32 [[I_NEXT]], [[N:%.*]]
; CHECK-NEXT: br i1 [[COND]], label [[LOOP]], label [[EXIT:%.*]]
; CHECK: exit:
; CHECK-NEXT: ret void
;
entry:
br label %loop

loop:
%i = phi i32 [ %i.init, %entry ], [ %i.next, %loop ]
%j = phi i32 [ %j.init, %entry ], [ %j.next, %loop ]
%k = phi i32 [ %k.init, %entry ], [ %any, %loop ]
%i.next = add nuw nsw i32 %i, 1
%j.next = add nuw nsw i32 %j, -2
%ij = add i32 %i, %j
%ijk = add i32 %ij, %k
%ijk.fr = freeze i32 %ijk
call void @call(i32 %ijk.fr)
%any = call i32 @call2()
%cond = icmp eq i32 %i.next, %n
br i1 %cond, label %loop, label %exit

exit:
ret void
}