| ; Test memset in cases where a loop is used. |
| ; |
| ; RUN: llc < %s -mtriple=s390x-linux-gnu | FileCheck %s |
| |
| declare void @llvm.memset.p0i8.i32(i8 *nocapture, i8, i32, i1) nounwind |
| declare void @llvm.memset.p0i8.i64(i8 *nocapture, i8, i64, i1) nounwind |
| |
| ; Constant length: 6 iterations and 2 bytes remainder. |
| define void @f1(i8* %dest, i8 %val) { |
| ; CHECK-LABEL: f1: |
| ; CHECK: lghi [[COUNT:%r[0-5]]], 6 |
| ; CHECK: [[LABEL:\.L[^:]*]]: |
| ; CHECK: pfd 2, 768(%r2) |
| ; CHECK: stc %r3, 0(%r2) |
| ; CHECK: mvc 1(255,%r2), 0(%r2) |
| ; CHECK: la %r2, 256(%r2) |
| ; CHECK: brctg [[COUNT]], [[LABEL]] |
| ; CHECK: stc %r3, 0(%r2) |
| ; CHECK-NEXT: mvc 1(1,%r2), 0(%r2) |
| ; CHECK-NEXT: br %r14 |
| call void @llvm.memset.p0i8.i64(i8* %dest, i8 %val, i64 1538, i1 false) |
| ret void |
| } |
| |
| ; Constant length: 6 iterations and 255 bytes remainder. |
| define void @f2(i8* %dest) { |
| ; CHECK-LABEL: f2: |
| ; CHECK: lghi [[COUNT:%r[0-5]]], 6 |
| ; CHECK: [[LABEL:\.L[^:]*]]: |
| ; CHECK: pfd 2, 768(%r2) |
| ; CHECK: mvi 0(%r2), 1 |
| ; CHECK: mvc 1(255,%r2), 0(%r2) |
| ; CHECK: la %r2, 256(%r2) |
| ; CHECK: brctg [[COUNT]], [[LABEL]] |
| ; CHECK: mvi 0(%r2), 1 |
| ; CHECK-NEXT: mvc 1(254,%r2), 0(%r2) |
| ; CHECK-NEXT: br %r14 |
| call void @llvm.memset.p0i8.i32(i8* %dest, i8 1, i32 1791, i1 false) |
| ret void |
| } |
| |
| ; Variable length, byte in register. |
| define void @f3(i8* %dest, i8 %val, i64 %Len) { |
| ; CHECK-LABEL: f3: |
| ; CHECK: # %bb.0: |
| ; CHECK-NEXT: aghi %r4, -2 |
| ; CHECK-NEXT: cgibe %r4, -2, 0(%r14) |
| ; CHECK-NEXT: .LBB2_1: |
| ; CHECK-NEXT: cgije %r4, -1, .LBB2_5 |
| ; CHECK-NEXT:# %bb.2: |
| ; CHECK-NEXT: srlg %r0, %r4, 8 |
| ; CHECK-NEXT: cgije %r0, 0, .LBB2_4 |
| ; CHECK-NEXT:.LBB2_3: # =>This Inner Loop Header: Depth=1 |
| ; CHECK-NEXT: pfd 2, 768(%r2) |
| ; CHECK-NEXT: stc %r3, 0(%r2) |
| ; CHECK-NEXT: mvc 1(255,%r2), 0(%r2) |
| ; CHECK-NEXT: la %r2, 256(%r2) |
| ; CHECK-NEXT: brctg %r0, .LBB2_3 |
| ; CHECK-NEXT:.LBB2_4: |
| ; CHECK-NEXT: stc %r3, 0(%r2) |
| ; CHECK-NEXT: exrl %r4, .Ltmp0 |
| ; CHECK-NEXT: br %r14 |
| ; CHECK-NEXT:.LBB2_5: |
| ; CHECK-NEXT: stc %r3, 0(%r2) |
| ; CHECK-NEXT: br %r14 |
| call void @llvm.memset.p0i8.i64(i8* %dest, i8 %val, i64 %Len, i1 false) |
| ret void |
| } |
| |
| ; Variable length, immediate byte. |
| define void @f4(i8* %dest, i32 %Len) { |
| ; CHECK-LABEL: f4: |
| ; CHECK: # %bb.0: |
| ; CHECK-NEXT: llgfr %r1, %r3 |
| ; CHECK-NEXT: aghi %r1, -2 |
| ; CHECK-NEXT: cgibe %r1, -2, 0(%r14) |
| ; CHECK-NEXT:.LBB3_1: |
| ; CHECK-NEXT: cgije %r1, -1, .LBB3_5 |
| ; CHECK-NEXT:# %bb.2: |
| ; CHECK-NEXT: srlg %r0, %r1, 8 |
| ; CHECK-NEXT: cgije %r0, 0, .LBB3_4 |
| ; CHECK-NEXT:.LBB3_3: # =>This Inner Loop Header: Depth=1 |
| ; CHECK-NEXT: pfd 2, 768(%r2) |
| ; CHECK-NEXT: mvi 0(%r2), 1 |
| ; CHECK-NEXT: mvc 1(255,%r2), 0(%r2) |
| ; CHECK-NEXT: la %r2, 256(%r2) |
| ; CHECK-NEXT: brctg %r0, .LBB3_3 |
| ; CHECK-NEXT:.LBB3_4: |
| ; CHECK-NEXT: mvi 0(%r2), 1 |
| ; CHECK-NEXT: exrl %r1, .Ltmp0 |
| ; CHECK-NEXT: br %r14 |
| ; CHECK-NEXT:.LBB3_5: |
| ; CHECK-NEXT: mvi 0(%r2), 1 |
| ; CHECK-NEXT: br %r14 |
| call void @llvm.memset.p0i8.i32(i8* %dest, i8 1, i32 %Len, i1 false) |
| ret void |
| } |
| |
| ; CHECK: .Ltmp0: |
| ; CHECK-NEXT: mvc 1(1,%r2), 0(%r2) |
