xref: /llvm-project/llvm/test/Transforms/IROutliner/outline-vaarg-intrinsic.ll (revision ab7dba233a058cc8310ef829929238b5d8440b30)
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --function-signature --include-generated-funcs
2; RUN: opt -S -passes=verify,iroutliner -ir-outlining-no-cost < %s | FileCheck %s
3
4; This test checks that we sucessfully outline identical memcpy var arg
5; intrinsics, but not the var arg instruction itself.
6
7declare void @llvm.va_start(ptr)
8declare void @llvm.va_copy(ptr, ptr)
9declare void @llvm.va_end(ptr)
10
11define i32 @func1(i32 %a, double %b, ptr %v, ...) nounwind {
12entry:
13  %a.addr = alloca i32, align 4
14  %b.addr = alloca double, align 8
15  %ap = alloca ptr, align 4
16  %c = alloca i32, align 4
17  store i32 %a, ptr %a.addr, align 4
18  store double %b, ptr %b.addr, align 8
19  call void @llvm.va_start(ptr %ap)
20  %0 = va_arg ptr %ap, i32
21  call void @llvm.va_copy(ptr %v, ptr %ap)
22  call void @llvm.va_end(ptr %ap)
23  store i32 %0, ptr %c, align 4
24  %tmp = load i32, ptr %c, align 4
25  ret i32 %tmp
26}
27
28define i32 @func2(i32 %a, double %b, ptr %v, ...) nounwind {
29entry:
30  %a.addr = alloca i32, align 4
31  %b.addr = alloca double, align 8
32  %ap = alloca ptr, align 4
33  %c = alloca i32, align 4
34  store i32 %a, ptr %a.addr, align 4
35  store double %b, ptr %b.addr, align 8
36  call void @llvm.va_start(ptr %ap)
37  %0 = va_arg ptr %ap, i32
38  call void @llvm.va_copy(ptr %v, ptr %ap)
39  call void @llvm.va_end(ptr %ap)
40  store i32 %0, ptr %c, align 4
41  %tmp = load i32, ptr %c, align 4
42  ret i32 %tmp
43}
44; CHECK-LABEL: define {{[^@]+}}@func1
45; CHECK-SAME: (i32 [[A:%.*]], double [[B:%.*]], ptr [[V:%.*]], ...) #[[ATTR1:[0-9]+]] {
46; CHECK-NEXT:  entry:
47; CHECK-NEXT:    [[TMP_LOC:%.*]] = alloca i32, align 4
48; CHECK-NEXT:    [[A_ADDR:%.*]] = alloca i32, align 4
49; CHECK-NEXT:    [[B_ADDR:%.*]] = alloca double, align 8
50; CHECK-NEXT:    [[AP:%.*]] = alloca ptr, align 4
51; CHECK-NEXT:    [[C:%.*]] = alloca i32, align 4
52; CHECK-NEXT:    store i32 [[A]], ptr [[A_ADDR]], align 4
53; CHECK-NEXT:    store double [[B]], ptr [[B_ADDR]], align 8
54; CHECK-NEXT:    call void @llvm.va_start.p0(ptr [[AP]])
55; CHECK-NEXT:    [[TMP0:%.*]] = va_arg ptr [[AP]], i32
56; CHECK-NEXT:    call void @llvm.lifetime.start.p0(i64 -1, ptr [[TMP_LOC]])
57; CHECK-NEXT:    call void @outlined_ir_func_0(ptr [[V]], ptr [[AP]], i32 [[TMP0]], ptr [[C]], ptr [[TMP_LOC]])
58; CHECK-NEXT:    [[TMP_RELOAD:%.*]] = load i32, ptr [[TMP_LOC]], align 4
59; CHECK-NEXT:    call void @llvm.lifetime.end.p0(i64 -1, ptr [[TMP_LOC]])
60; CHECK-NEXT:    ret i32 [[TMP_RELOAD]]
61;
62;
63; CHECK-LABEL: define {{[^@]+}}@func2
64; CHECK-SAME: (i32 [[A:%.*]], double [[B:%.*]], ptr [[V:%.*]], ...) #[[ATTR1]] {
65; CHECK-NEXT:  entry:
66; CHECK-NEXT:    [[TMP_LOC:%.*]] = alloca i32, align 4
67; CHECK-NEXT:    [[A_ADDR:%.*]] = alloca i32, align 4
68; CHECK-NEXT:    [[B_ADDR:%.*]] = alloca double, align 8
69; CHECK-NEXT:    [[AP:%.*]] = alloca ptr, align 4
70; CHECK-NEXT:    [[C:%.*]] = alloca i32, align 4
71; CHECK-NEXT:    store i32 [[A]], ptr [[A_ADDR]], align 4
72; CHECK-NEXT:    store double [[B]], ptr [[B_ADDR]], align 8
73; CHECK-NEXT:    call void @llvm.va_start.p0(ptr [[AP]])
74; CHECK-NEXT:    [[TMP0:%.*]] = va_arg ptr [[AP]], i32
75; CHECK-NEXT:    call void @llvm.lifetime.start.p0(i64 -1, ptr [[TMP_LOC]])
76; CHECK-NEXT:    call void @outlined_ir_func_0(ptr [[V]], ptr [[AP]], i32 [[TMP0]], ptr [[C]], ptr [[TMP_LOC]])
77; CHECK-NEXT:    [[TMP_RELOAD:%.*]] = load i32, ptr [[TMP_LOC]], align 4
78; CHECK-NEXT:    call void @llvm.lifetime.end.p0(i64 -1, ptr [[TMP_LOC]])
79; CHECK-NEXT:    ret i32 [[TMP_RELOAD]]
80;
81;
82; CHECK-LABEL: define {{[^@]+}}@outlined_ir_func_0
83; CHECK-SAME: (ptr [[TMP0:%.*]], ptr [[TMP1:%.*]], i32 [[TMP2:%.*]], ptr [[TMP3:%.*]], ptr [[TMP4:%.*]]) #[[ATTR3:[0-9]+]] {
84; CHECK-NEXT:  newFuncRoot:
85; CHECK-NEXT:    br label [[ENTRY_TO_OUTLINE:%.*]]
86; CHECK:       entry_to_outline:
87; CHECK-NEXT:    call void @llvm.va_copy.p0(ptr [[TMP0]], ptr [[TMP1]])
88; CHECK-NEXT:    call void @llvm.va_end.p0(ptr [[TMP1]])
89; CHECK-NEXT:    store i32 [[TMP2]], ptr [[TMP3]], align 4
90; CHECK-NEXT:    [[TMP:%.*]] = load i32, ptr [[TMP3]], align 4
91; CHECK-NEXT:    br label [[ENTRY_AFTER_OUTLINE_EXITSTUB:%.*]]
92; CHECK:       entry_after_outline.exitStub:
93; CHECK-NEXT:    store i32 [[TMP]], ptr [[TMP4]], align 4
94; CHECK-NEXT:    ret void
95;
96