test/CodeGen/SystemZ/memset-01.ll

   1 ; Test memset in cases where the set value is variable.
   2 ;
   3 ; RUN: llc < %s -mtriple=s390x-linux-gnu | FileCheck %s
   4
   5 declare void @llvm.memset.p0i8.i32(i8 *nocapture, i8, i32, i1) nounwind
   6 declare void @llvm.memset.p0i8.i64(i8 *nocapture, i8, i64, i1) nounwind
   7
   8 ; No bytes, i32 version.
   9 define 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, i1 false)
  15   ret void
  16 }
  17
  18 ; No bytes, i64 version.
  19 define 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, i1 false)
  25   ret void
  26 }
  27
  28 ; 1 byte, i32 version.
  29 define 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, i1 false)
  34   ret void
  35 }
  36
  37 ; 1 byte, i64 version.
  38 define 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, i1 false)
  43   ret void
  44 }
  45
  46 ; 2 bytes, i32 version.
  47 define 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, i1 false)
  53   ret void
  54 }
  55
  56 ; 2 bytes, i64 version.
  57 define 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, i1 false)
  63   ret void
  64 }
  65
  66 ; 3 bytes, i32 version.
  67 define 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, i1 false)
  73   ret void
  74 }
  75
  76 ; 3 bytes, i64 version.
  77 define 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, i1 false)
  83   ret void
  84 }
  85
  86 ; 257 bytes, i32 version.
  87 define 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, i1 false)
  93   ret void
  94 }
  95
  96 ; 257 bytes, i64 version.
  97 define 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, i1 false)
 103   ret void
 104 }
 105
 106 ; 258 bytes, i32 version.  We need two MVCs.
 107 define 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, i1 false)
 114   ret void
 115 }
 116
 117 ; 258 bytes, i64 version.
 118 define 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, i1 false)
 125   ret void
 126 }
 127
 128 ; Test the largest case for which straight-line code is used.
 129 define 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, i1 false)
 140   ret void
 141 }
 142
 143 ; Test the next size up, which uses a loop.  We leave the other corner
 144 ; cases to memcpy-01.ll.
 145 define void @f14(i8* %dest, i8 %val) {
 146 ; CHECK-LABEL: f14:
 147 ; CHECK: stc %r3, 0(%r2)
 148 ; CHECK: lghi [[COUNT:%r[0-5]]], 6
 149 ; CHECK: [[LABEL:\.L[^:]*]]:
 150 ; CHECK: pfd 2, 769(%r2)
 151 ; CHECK: mvc 1(256,%r2), 0(%r2)
 152 ; CHECK: la %r2, 256(%r2)
 153 ; CHECK: brctg [[COUNT]], [[LABEL]]
 154 ; CHECK: mvc 1(1,%r2), 0(%r2)
 155 ; CHECK: br %r14
 156   call void @llvm.memset.p0i8.i64(i8* %dest, i8 %val, i64 1538, i1 false)
 157   ret void
 158 }