1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
; NOTE: Assertions have been autogenerated by utils/update_analyze_test_checks.py UTC_ARGS: --version 5
; REQUIRES: asserts
; RUN: opt -passes=loop-distribute -enable-loop-distribute \
; RUN: -debug-only=loop-distribute -disable-output 2>&1 %s | FileCheck %s
define void @f(ptr noalias %a, ptr noalias %b, ptr noalias %c, ptr noalias %d, i64 %stride) {
; CHECK-LABEL: 'f'
; CHECK-NEXT: LDist: Found a candidate loop: for.body
; CHECK-NEXT: LDist: Backward dependences:
; CHECK-NEXT: Backward:
; CHECK-NEXT: %load.a = load i32, ptr %gep.a, align 4 ->
; CHECK-NEXT: store i32 %mul.a, ptr %gep.a.plus4, align 4
; CHECK-NEXT: LDist: Seeded partitions:
; CHECK-NEXT: LDist: Partition 0: (cycle)
; CHECK-NEXT: for.body: %load.a = load i32, ptr %gep.a, align 4
; CHECK-NEXT: for.body: %load.b = load i32, ptr %gep.b, align 4
; CHECK-NEXT: for.body: store i32 %mul.a, ptr %gep.a.plus4, align 4
; CHECK-NEXT: LDist: Partition 1:
; CHECK-NEXT: for.body: %loadD = load i32, ptr %gep.d, align 4
; CHECK-NEXT: LDist: Partition 2:
; CHECK-NEXT: for.body: %load.strided.a = load i32, ptr %gep.strided.a, align 4
; CHECK-NEXT: LDist: Partition 3:
; CHECK-NEXT: for.body: store i32 %mul.c, ptr %gep.c, align 4
; CHECK-NEXT: LDist: Merged partitions:
; CHECK-NEXT: LDist: Partition 0: (cycle)
; CHECK-NEXT: for.body: %load.a = load i32, ptr %gep.a, align 4
; CHECK-NEXT: for.body: %load.b = load i32, ptr %gep.b, align 4
; CHECK-NEXT: for.body: store i32 %mul.a, ptr %gep.a.plus4, align 4
; CHECK-NEXT: LDist: Partition 1:
; CHECK-NEXT: for.body: %loadD = load i32, ptr %gep.d, align 4
; CHECK-NEXT: for.body: %load.strided.a = load i32, ptr %gep.strided.a, align 4
; CHECK-NEXT: for.body: store i32 %mul.c, ptr %gep.c, align 4
; CHECK-NEXT: LDist: Populated partitions:
; CHECK-NEXT: LDist: Partition 0: (cycle)
; CHECK-NEXT: for.body: %load.a = load i32, ptr %gep.a, align 4
; CHECK-NEXT: for.body: %load.b = load i32, ptr %gep.b, align 4
; CHECK-NEXT: for.body: store i32 %mul.a, ptr %gep.a.plus4, align 4
; CHECK-NEXT: for.body: br i1 %exitcond, label %exit, label %for.body
; CHECK-NEXT: for.body: %exitcond = icmp eq i64 %add, 20
; CHECK-NEXT: for.body: %add = add nuw nsw i64 %ind, 1
; CHECK-NEXT: for.body: %ind = phi i64 [ 0, %entry ], [ %add, %for.body ]
; CHECK-NEXT: for.body: %mul.a = mul i32 %load.b, %load.a
; CHECK-NEXT: for.body: %gep.a.plus4 = getelementptr inbounds i32, ptr %a, i64 %add
; CHECK-NEXT: for.body: %gep.b = getelementptr inbounds i32, ptr %b, i64 %ind
; CHECK-NEXT: for.body: %gep.a = getelementptr inbounds i32, ptr %a, i64 %ind
; CHECK-NEXT: LDist: Partition 1:
; CHECK-NEXT: for.body: %loadD = load i32, ptr %gep.d, align 4
; CHECK-NEXT: for.body: %load.strided.a = load i32, ptr %gep.strided.a, align 4
; CHECK-NEXT: for.body: store i32 %mul.c, ptr %gep.c, align 4
; CHECK-NEXT: for.body: br i1 %exitcond, label %exit, label %for.body
; CHECK-NEXT: for.body: %exitcond = icmp eq i64 %add, 20
; CHECK-NEXT: for.body: %add = add nuw nsw i64 %ind, 1
; CHECK-NEXT: for.body: %ind = phi i64 [ 0, %entry ], [ %add, %for.body ]
; CHECK-NEXT: for.body: %mul.c = mul i32 %loadD, %load.strided.a
; CHECK-NEXT: for.body: %gep.c = getelementptr inbounds i32, ptr %c, i64 %ind
; CHECK-NEXT: for.body: %gep.strided.a = getelementptr inbounds i32, ptr %a, i64 %mul
; CHECK-NEXT: for.body: %mul = mul i64 %ind, %stride
; CHECK-NEXT: for.body: %gep.d = getelementptr inbounds i32, ptr %d, i64 %ind
; CHECK-NEXT: LDist: Distributing loop: for.body
; CHECK-NEXT: LDist: Pointers:
; CHECK-NEXT: LDist: After removing unused Instrs:
; CHECK-NEXT: LDist: Partition 0:
; CHECK-NEXT: for.body.ldist1: ; preds = %for.body.ldist1, %for.body.ph.ldist1
; CHECK-NEXT: %ind.ldist1 = phi i64 [ 0, %for.body.ph.ldist1 ], [ %add.ldist1, %for.body.ldist1 ]
; CHECK-NEXT: %gep.a.ldist1 = getelementptr inbounds i32, ptr %a, i64 %ind.ldist1
; CHECK-NEXT: %load.a.ldist1 = load i32, ptr %gep.a.ldist1, align 4
; CHECK-NEXT: %gep.b.ldist1 = getelementptr inbounds i32, ptr %b, i64 %ind.ldist1
; CHECK-NEXT: %load.b.ldist1 = load i32, ptr %gep.b.ldist1, align 4
; CHECK-NEXT: %mul.a.ldist1 = mul i32 %load.b.ldist1, %load.a.ldist1
; CHECK-NEXT: %add.ldist1 = add nuw nsw i64 %ind.ldist1, 1
; CHECK-NEXT: %gep.a.plus4.ldist1 = getelementptr inbounds i32, ptr %a, i64 %add.ldist1
; CHECK-NEXT: store i32 %mul.a.ldist1, ptr %gep.a.plus4.ldist1, align 4
; CHECK-NEXT: %exitcond.ldist1 = icmp eq i64 %add.ldist1, 20
; CHECK-NEXT: br i1 %exitcond.ldist1, label %for.body.ph, label %for.body.ldist1
; CHECK-NEXT: LDist: Partition 1:
; CHECK-NEXT: for.body: ; preds = %for.body, %for.body.ph
; CHECK-NEXT: %ind = phi i64 [ 0, %for.body.ph ], [ %add, %for.body ]
; CHECK-NEXT: %add = add nuw nsw i64 %ind, 1
; CHECK-NEXT: %gep.d = getelementptr inbounds i32, ptr %d, i64 %ind
; CHECK-NEXT: %loadD = load i32, ptr %gep.d, align 4
; CHECK-NEXT: %mul = mul i64 %ind, %stride
; CHECK-NEXT: %gep.strided.a = getelementptr inbounds i32, ptr %a, i64 %mul
; CHECK-NEXT: %load.strided.a = load i32, ptr %gep.strided.a, align 4
; CHECK-NEXT: %mul.c = mul i32 %loadD, %load.strided.a
; CHECK-NEXT: %gep.c = getelementptr inbounds i32, ptr %c, i64 %ind
; CHECK-NEXT: store i32 %mul.c, ptr %gep.c, align 4
; CHECK-NEXT: %exitcond = icmp eq i64 %add, 20
; CHECK-NEXT: br i1 %exitcond, label %exit.loopexit1, label %for.body
;
entry:
br label %for.body
for.body: ; preds = %for.body, %entry
%ind = phi i64 [ 0, %entry ], [ %add, %for.body ]
%gep.a = getelementptr inbounds i32, ptr %a, i64 %ind
%load.a = load i32, ptr %gep.a, align 4
%gep.b = getelementptr inbounds i32, ptr %b, i64 %ind
%load.b = load i32, ptr %gep.b, align 4
%mul.a = mul i32 %load.b, %load.a
%add = add nuw nsw i64 %ind, 1
%gep.a.plus4 = getelementptr inbounds i32, ptr %a, i64 %add
store i32 %mul.a, ptr %gep.a.plus4, align 4
%gep.d = getelementptr inbounds i32, ptr %d, i64 %ind
%loadD = load i32, ptr %gep.d, align 4
%mul = mul i64 %ind, %stride
%gep.strided.a = getelementptr inbounds i32, ptr %a, i64 %mul
%load.strided.a = load i32, ptr %gep.strided.a, align 4
%mul.c = mul i32 %loadD, %load.strided.a
%gep.c = getelementptr inbounds i32, ptr %c, i64 %ind
store i32 %mul.c, ptr %gep.c, align 4
%exitcond = icmp eq i64 %add, 20
br i1 %exitcond, label %exit, label %for.body
exit: ; preds = %for.body
ret void
}
|
