blob: 19198dd3c3b71f252f016853a72b297bf2dd94e5 [file] [log] [blame]
; Test memset in cases where a loop is used.
;
; RUN: llc < %s -mtriple=s390x-linux-gnu | FileCheck %s
declare void @llvm.memset.p0.i32(ptr nocapture, i8, i32, i1) nounwind
declare void @llvm.memset.p0.i64(ptr nocapture, i8, i64, i1) nounwind
; Constant length: 6 iterations and 2 bytes remainder.
define void @f1(ptr %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.p0.i64(ptr %dest, i8 %val, i64 1538, i1 false)
ret void
}
; Constant length: 6 iterations and 255 bytes remainder.
define void @f2(ptr %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.p0.i32(ptr %dest, i8 1, i32 1791, i1 false)
ret void
}
; Variable length, byte in register.
define void @f3(ptr %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.p0.i64(ptr %dest, i8 %val, i64 %Len, i1 false)
ret void
}
; Variable length, immediate byte.
define void @f4(ptr %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.p0.i32(ptr %dest, i8 1, i32 %Len, i1 false)
ret void
}
; CHECK: .Ltmp0:
; CHECK-NEXT: mvc 1(1,%r2), 0(%r2)