1; RUN: llc -mtriple=s390x-linux-gnu < %s | FileCheck %s
2
3; Trivial patchpoint codegen
4;
5define i64 @trivial_patchpoint_codegen(i64 %p1, i64 %p2, i64 %p3, i64 %p4) {
6entry:
7; CHECK-LABEL: trivial_patchpoint_codegen:
8; CHECK:       llilf   %r1, 559038736
9; CHECK-NEXT:  basr    %r14, %r1
10; CHECK-NEXT:  bcr     0, %r0
11; CHECK:       lgr     [[REG0:%r[0-9]+]], %r2
12; CHECK:       llilf   %r1, 559038737
13; CHECK-NEXT:  basr    %r14, %r1
14; CHECK-NEXT:  bcr     0, %r0
15; CHECK:       lgr     %r2, [[REG0:%r[0-9]+]]
16; CHECK:       br      %r14
17  %resolveCall2 = inttoptr i64 559038736 to i8*
18  %result = tail call i64 (i64, i32, i8*, i32, ...) @llvm.experimental.patchpoint.i64(i64 2, i32 10, i8* %resolveCall2, i32 4, i64 %p1, i64 %p2, i64 %p3, i64 %p4)
19  %resolveCall3 = inttoptr i64 559038737 to i8*
20  tail call void (i64, i32, i8*, i32, ...) @llvm.experimental.patchpoint.void(i64 3, i32 10, i8* %resolveCall3, i32 2, i64 %p1, i64 %result)
21  ret i64 %result
22}
23
24; Trivial symbolic patchpoint codegen.
25;
26
27declare i64 @foo(i64 %p1, i64 %p2)
28define i64 @trivial_symbolic_patchpoint_codegen(i64 %p1, i64 %p2) {
29entry:
30; CHECK-LABEL: trivial_symbolic_patchpoint_codegen:
31; CHECK:       brasl   %r14, foo@PLT
32; CHECK-NEXT:  bcr     0, %r0
33; CHECK:       br      %r14
34  %result = tail call i64 (i64, i32, i8*, i32, ...) @llvm.experimental.patchpoint.i64(i64 9, i32 8, i8* bitcast (i64 (i64, i64)* @foo to i8*), i32 2, i64 %p1, i64 %p2)
35  ret i64 %result
36}
37
38
39; Caller frame metadata with stackmaps. This should not be optimized
40; as a leaf function.
41;
42; CHECK-LABEL: caller_meta_leaf
43; CHECK: aghi  %r15, -184
44; CHECK: .Ltmp
45; CHECK: lmg   %r14, %r15, 296(%r15)
46; CHECK: br    %r14
47define void @caller_meta_leaf() {
48entry:
49  %metadata = alloca i64, i32 3, align 8
50  store i64 11, i64* %metadata
51  store i64 12, i64* %metadata
52  store i64 13, i64* %metadata
53  call void (i64, i32, ...) @llvm.experimental.stackmap(i64 4, i32 0, i64* %metadata)
54  ret void
55}
56
57; Test patchpoints reusing the same TargetConstant.
58; <rdar:15390785> Assertion failed: (CI.getNumArgOperands() >= NumArgs + 4)
59; There is no way to verify this, since it depends on memory allocation.
60; But I think it's useful to include as a working example.
61define i64 @testLowerConstant(i64 %arg, i64 %tmp2, i64 %tmp10, i64* %tmp33, i64 %tmp79) {
62entry:
63  %tmp80 = add i64 %tmp79, -16
64  %tmp81 = inttoptr i64 %tmp80 to i64*
65  %tmp82 = load i64, i64* %tmp81, align 8
66  tail call void (i64, i32, ...) @llvm.experimental.stackmap(i64 14, i32 6, i64 %arg, i64 %tmp2, i64 %tmp10, i64 %tmp82)
67  tail call void (i64, i32, i8*, i32, ...) @llvm.experimental.patchpoint.void(i64 15, i32 30, i8* null, i32 3, i64 %arg, i64 %tmp10, i64 %tmp82)
68  %tmp83 = load i64, i64* %tmp33, align 8
69  %tmp84 = add i64 %tmp83, -24
70  %tmp85 = inttoptr i64 %tmp84 to i64*
71  %tmp86 = load i64, i64* %tmp85, align 8
72  tail call void (i64, i32, ...) @llvm.experimental.stackmap(i64 17, i32 6, i64 %arg, i64 %tmp10, i64 %tmp86)
73  tail call void (i64, i32, i8*, i32, ...) @llvm.experimental.patchpoint.void(i64 18, i32 30, i8* null, i32 3, i64 %arg, i64 %tmp10, i64 %tmp86)
74  ret i64 10
75}
76
77; Test small patchpoints that don't emit calls.
78define void @small_patchpoint_codegen(i64 %p1, i64 %p2, i64 %p3, i64 %p4) {
79entry:
80; CHECK-LABEL: small_patchpoint_codegen:
81; CHECK:      .Ltmp
82; CHECK:      bcr 0, %r0
83; CHECK:      br %r14
84  %result = tail call i64 (i64, i32, i8*, i32, ...) @llvm.experimental.patchpoint.i64(i64 5, i32 2, i8* null, i32 2, i64 %p1, i64 %p2)
85  ret void
86}
87
88; Test large target address.
89define i64 @large_target_address_patchpoint_codegen() {
90entry:
91; CHECK-LABEL: large_target_address_patchpoint_codegen:
92; CHECK:        llilf   %r1, 2566957755
93; CHECK-NEXT:   iihf    %r1, 1432778632
94; CHECK-NEXT:   basr    %r14, %r1
95  %resolveCall2 = inttoptr i64 6153737369414576827 to i8*
96  %result = tail call i64 (i64, i32, i8*, i32, ...) @llvm.experimental.patchpoint.i64(i64 2, i32 14, i8* %resolveCall2, i32 0)
97  ret i64 %result
98}
99
100declare void @llvm.experimental.stackmap(i64, i32, ...)
101declare void @llvm.experimental.patchpoint.void(i64, i32, i8*, i32, ...)
102declare i64 @llvm.experimental.patchpoint.i64(i64, i32, i8*, i32, ...)
103