caio/clang-p2996
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[SROA] Add load-only promotion tests with dynamic offset load

Browse Source
This commit is contained in:
Nikita Popov
2025-04-14 12:15:16 +02:00
parent a9474191e0
commit 1e2dc5b087
1 changed files with 80 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

80
llvm/test/Transforms/SROA/readonlynocapture.ll
View File

@@ -499,4 +499,84 @@ define i32 @twoalloc_with_lifetimes() {
ret i32 %r
}
declare void @use.i32(i32)
define void @load_dyn_offset(ptr %ary) {
; CHECK-LABEL: @load_dyn_offset(
; CHECK-NEXT: [[A:%.*]] = alloca { i64, [4 x i32] }, align 8
; CHECK-NEXT: store i64 0, ptr [[A]], align 4
; CHECK-NEXT: [[GEP:%.*]] = getelementptr i8, ptr [[A]], i64 8
; CHECK-NEXT: call void @llvm.memcpy.p0.p0.i64(ptr [[GEP]], ptr [[ARY:%.*]], i64 16, i1 false)
; CHECK-NEXT: br label [[LOOP:%.*]]
; CHECK: loop:
; CHECK-NEXT: [[I:%.*]] = load i64, ptr [[A]], align 4
; CHECK-NEXT: [[I_NEXT:%.*]] = add i64 [[I]], 1
; CHECK-NEXT: store i64 [[I_NEXT]], ptr [[A]], align 4
; CHECK-NEXT: [[GEP_I:%.*]] = getelementptr i32, ptr [[GEP]], i64 [[I]]
; CHECK-NEXT: [[VAL:%.*]] = load i32, ptr [[GEP_I]], align 4
; CHECK-NEXT: call void @use.i32(i32 [[VAL]])
; CHECK-NEXT: [[CMP:%.*]] = icmp eq i64 [[I_NEXT]], 6
; CHECK-NEXT: br i1 [[CMP]], label [[EXIT:%.*]], label [[LOOP]]
; CHECK: exit:
; CHECK-NEXT: ret void
;
%a = alloca {i64, [4 x i32]}
store i64 0, ptr %a
%gep = getelementptr i8, ptr %a, i64 8
call void @llvm.memcpy(ptr %gep, ptr %ary, i64 16, i1 false)
br label %loop
loop:
%i = load i64, ptr %a
%i.next = add i64 %i, 1
store i64 %i.next, ptr %a
%gep.i = getelementptr i32, ptr %gep, i64 %i
%val = load i32, ptr %gep.i
call void @use.i32(i32 %val)
%cmp = icmp eq i64 %i.next, 6
br i1 %cmp, label %exit, label %loop
exit:
ret void
}
define void @store_dyn_offset(ptr %ary) {
; CHECK-LABEL: @store_dyn_offset(
; CHECK-NEXT: [[A:%.*]] = alloca { i64, [4 x i32] }, align 8
; CHECK-NEXT: store i64 0, ptr [[A]], align 4
; CHECK-NEXT: [[GEP:%.*]] = getelementptr i8, ptr [[A]], i64 8
; CHECK-NEXT: call void @llvm.memcpy.p0.p0.i64(ptr [[GEP]], ptr [[ARY:%.*]], i64 16, i1 false)
; CHECK-NEXT: br label [[LOOP:%.*]]
; CHECK: loop:
; CHECK-NEXT: [[I:%.*]] = load i64, ptr [[A]], align 4
; CHECK-NEXT: [[I_NEXT:%.*]] = add i64 [[I]], 1
; CHECK-NEXT: store i64 [[I_NEXT]], ptr [[A]], align 4
; CHECK-NEXT: [[GEP_I:%.*]] = getelementptr i32, ptr [[GEP]], i64 [[I]]
; CHECK-NEXT: [[I_TRUNC:%.*]] = trunc i64 [[I]] to i32
; CHECK-NEXT: store i32 [[I_TRUNC]], ptr [[GEP_I]], align 4
; CHECK-NEXT: [[CMP:%.*]] = icmp eq i64 [[I_NEXT]], 6
; CHECK-NEXT: br i1 [[CMP]], label [[EXIT:%.*]], label [[LOOP]]
; CHECK: exit:
; CHECK-NEXT: ret void
;
%a = alloca {i64, [4 x i32]}
store i64 0, ptr %a
%gep = getelementptr i8, ptr %a, i64 8
call void @llvm.memcpy(ptr %gep, ptr %ary, i64 16, i1 false)
br label %loop
loop:
%i = load i64, ptr %a
%i.next = add i64 %i, 1
store i64 %i.next, ptr %a
%gep.i = getelementptr i32, ptr %gep, i64 %i
%i.trunc = trunc i64 %i to i32
store i32 %i.trunc, ptr %gep.i
%cmp = icmp eq i64 %i.next, 6
br i1 %cmp, label %exit, label %loop
exit:
ret void
}
declare void @llvm.memcpy.p0.p0.i64(ptr, ptr, i64, i1)