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