1; Test memset in cases where the set value is variable.
2;
3; RUN: llc < %s -mtriple=s390x-linux-gnu | FileCheck %s
4
5declare void @llvm.memset.p0i8.i32(i8 *nocapture, i8, i32, i32, i1) nounwind
6declare void @llvm.memset.p0i8.i64(i8 *nocapture, i8, i64, i32, i1) nounwind
7
8; No bytes, i32 version.
9define void @f1(i8 *%dest, i8 %val) {
10; CHECK-LABEL: f1:
11; CHECK-NOT: %r2
12; CHECK-NOT: %r3
13; CHECK: br %r14
14  call void @llvm.memset.p0i8.i32(i8 *%dest, i8 %val, i32 0, i32 1, i1 false)
15  ret void
16}
17
18; No bytes, i64 version.
19define void @f2(i8 *%dest, i8 %val) {
20; CHECK-LABEL: f2:
21; CHECK-NOT: %r2
22; CHECK-NOT: %r3
23; CHECK: br %r14
24  call void @llvm.memset.p0i8.i64(i8 *%dest, i8 %val, i64 0, i32 1, i1 false)
25  ret void
26}
27
28; 1 byte, i32 version.
29define void @f3(i8 *%dest, i8 %val) {
30; CHECK-LABEL: f3:
31; CHECK: stc %r3, 0(%r2)
32; CHECK: br %r14
33  call void @llvm.memset.p0i8.i32(i8 *%dest, i8 %val, i32 1, i32 1, i1 false)
34  ret void
35}
36
37; 1 byte, i64 version.
38define void @f4(i8 *%dest, i8 %val) {
39; CHECK-LABEL: f4:
40; CHECK: stc %r3, 0(%r2)
41; CHECK: br %r14
42  call void @llvm.memset.p0i8.i64(i8 *%dest, i8 %val, i64 1, i32 1, i1 false)
43  ret void
44}
45
46; 2 bytes, i32 version.
47define void @f5(i8 *%dest, i8 %val) {
48; CHECK-LABEL: f5:
49; CHECK-DAG: stc %r3, 0(%r2)
50; CHECK-DAG: stc %r3, 1(%r2)
51; CHECK: br %r14
52  call void @llvm.memset.p0i8.i32(i8 *%dest, i8 %val, i32 2, i32 1, i1 false)
53  ret void
54}
55
56; 2 bytes, i64 version.
57define void @f6(i8 *%dest, i8 %val) {
58; CHECK-LABEL: f6:
59; CHECK-DAG: stc %r3, 0(%r2)
60; CHECK-DAG: stc %r3, 1(%r2)
61; CHECK: br %r14
62  call void @llvm.memset.p0i8.i64(i8 *%dest, i8 %val, i64 2, i32 1, i1 false)
63  ret void
64}
65
66; 3 bytes, i32 version.
67define void @f7(i8 *%dest, i8 %val) {
68; CHECK-LABEL: f7:
69; CHECK: stc %r3, 0(%r2)
70; CHECK: mvc 1(2,%r2), 0(%r2)
71; CHECK: br %r14
72  call void @llvm.memset.p0i8.i32(i8 *%dest, i8 %val, i32 3, i32 1, i1 false)
73  ret void
74}
75
76; 3 bytes, i64 version.
77define void @f8(i8 *%dest, i8 %val) {
78; CHECK-LABEL: f8:
79; CHECK: stc %r3, 0(%r2)
80; CHECK: mvc 1(2,%r2), 0(%r2)
81; CHECK: br %r14
82  call void @llvm.memset.p0i8.i64(i8 *%dest, i8 %val, i64 3, i32 1, i1 false)
83  ret void
84}
85
86; 257 bytes, i32 version.
87define void @f9(i8 *%dest, i8 %val) {
88; CHECK-LABEL: f9:
89; CHECK: stc %r3, 0(%r2)
90; CHECK: mvc 1(256,%r2), 0(%r2)
91; CHECK: br %r14
92  call void @llvm.memset.p0i8.i32(i8 *%dest, i8 %val, i32 257, i32 1, i1 false)
93  ret void
94}
95
96; 257 bytes, i64 version.
97define void @f10(i8 *%dest, i8 %val) {
98; CHECK-LABEL: f10:
99; CHECK: stc %r3, 0(%r2)
100; CHECK: mvc 1(256,%r2), 0(%r2)
101; CHECK: br %r14
102  call void @llvm.memset.p0i8.i64(i8 *%dest, i8 %val, i64 257, i32 1, i1 false)
103  ret void
104}
105
106; 258 bytes, i32 version.  We need two MVCs.
107define void @f11(i8 *%dest, i8 %val) {
108; CHECK-LABEL: f11:
109; CHECK: stc %r3, 0(%r2)
110; CHECK: mvc 1(256,%r2), 0(%r2)
111; CHECK: mvc 257(1,%r2), 256(%r2)
112; CHECK: br %r14
113  call void @llvm.memset.p0i8.i32(i8 *%dest, i8 %val, i32 258, i32 1, i1 false)
114  ret void
115}
116
117; 258 bytes, i64 version.
118define void @f12(i8 *%dest, i8 %val) {
119; CHECK-LABEL: f12:
120; CHECK: stc %r3, 0(%r2)
121; CHECK: mvc 1(256,%r2), 0(%r2)
122; CHECK: mvc 257(1,%r2), 256(%r2)
123; CHECK: br %r14
124  call void @llvm.memset.p0i8.i64(i8 *%dest, i8 %val, i64 258, i32 1, i1 false)
125  ret void
126}
127
128; Test the largest case for which straight-line code is used.
129define void @f13(i8 *%dest, i8 %val) {
130; CHECK-LABEL: f13:
131; CHECK: stc %r3, 0(%r2)
132; CHECK: mvc 1(256,%r2), 0(%r2)
133; CHECK: mvc 257(256,%r2), 256(%r2)
134; CHECK: mvc 513(256,%r2), 512(%r2)
135; CHECK: mvc 769(256,%r2), 768(%r2)
136; CHECK: mvc 1025(256,%r2), 1024(%r2)
137; CHECK: mvc 1281(256,%r2), 1280(%r2)
138; CHECK: br %r14
139  call void @llvm.memset.p0i8.i64(i8 *%dest, i8 %val, i64 1537, i32 1,
140                                  i1 false)
141  ret void
142}
143
144; Test the next size up, which uses a loop.  We leave the other corner
145; cases to memcpy-01.ll.
146define void @f14(i8 *%dest, i8 %val) {
147; CHECK-LABEL: f14:
148; CHECK: stc %r3, 0(%r2)
149; CHECK: lghi [[COUNT:%r[0-5]]], 6
150; CHECK: [[LABEL:\.L[^:]*]]:
151; CHECK: pfd 2, 769(%r2)
152; CHECK: mvc 1(256,%r2), 0(%r2)
153; CHECK: la %r2, 256(%r2)
154; CHECK: brctg [[COUNT]], [[LABEL]]
155; CHECK: mvc 1(1,%r2), 0(%r2)
156; CHECK: br %r14
157  call void @llvm.memset.p0i8.i64(i8 *%dest, i8 %val, i64 1538, i32 1,
158                                  i1 false)
159  ret void
160}
161