; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
; Test memset 0 with variable length
;
; RUN: llc < %s -mtriple=s390x-linux-gnu | FileCheck %s

define void @fun0(i8* %Addr, i64 %Len) {
; CHECK-LABEL: fun0:
; CHECK:       # %bb.0:
; CHECK-NEXT:    aghi %r3, -1
; CHECK-NEXT:    cgibe %r3, -1, 0(%r14)
; CHECK-NEXT:  .LBB0_1:
; CHECK-NEXT:    srlg %r0, %r3, 8
; CHECK-NEXT:    cgije %r0, 0, .LBB0_3
; CHECK-NEXT:  .LBB0_2: # =>This Inner Loop Header: Depth=1
; CHECK-NEXT:    xc 0(256,%r2), 0(%r2)
; CHECK-NEXT:    la %r2, 256(%r2)
; CHECK-NEXT:    brctg %r0, .LBB0_2
; CHECK-NEXT:  .LBB0_3:
; CHECK-NEXT:    exrl %r3, .Ltmp0
; CHECK-NEXT:    br %r14
  tail call void @llvm.memset.p0i8.i64(i8* %Addr, i8 0, i64 %Len, i1 false)
  ret void
}

define void @fun1(i8* %Addr, i32 %Len) {
; CHECK-LABEL: fun1:
; CHECK:       # %bb.0:
; CHECK-NEXT:    llgfr %r1, %r3
; CHECK-NEXT:    aghi %r1, -1
; CHECK-NEXT:    cgibe %r1, -1, 0(%r14)
; CHECK-NEXT:  .LBB1_1:
; CHECK-NEXT:    srlg %r0, %r1, 8
; CHECK-NEXT:    cgije %r0, 0, .LBB1_3
; CHECK-NEXT:  .LBB1_2: # =>This Inner Loop Header: Depth=1
; CHECK-NEXT:    xc 0(256,%r2), 0(%r2)
; CHECK-NEXT:    la %r2, 256(%r2)
; CHECK-NEXT:    brctg %r0, .LBB1_2
; CHECK-NEXT:  .LBB1_3:
; CHECK-NEXT:    exrl %r1, .Ltmp0
; CHECK-NEXT:    br %r14
  tail call void @llvm.memset.p0i8.i32(i8* %Addr, i8 0, i32 %Len, i1 false)
  ret void
}

; Test that identical target instructions get reused.
define void @fun2(i8* %Addr, i32 %Len) {
; CHECK-LABEL: fun2:
; CHECK:       # %bb.0:
; CHECK-NEXT:    llgfr %r1, %r3
; CHECK-NEXT:    aghi %r1, -1
; CHECK-NEXT:    cgije %r1, -1, .LBB2_4
; CHECK-NEXT:  # %bb.1:
; CHECK-NEXT:    srlg %r0, %r1, 8
; CHECK-NEXT:    lgr %r3, %r2
; CHECK-NEXT:    cgije %r0, 0, .LBB2_3
; CHECK-NEXT:  .LBB2_2: # =>This Inner Loop Header: Depth=1
; CHECK-NEXT:    xc 0(256,%r3), 0(%r3)
; CHECK-NEXT:    la %r3, 256(%r3)
; CHECK-NEXT:    brctg %r0, .LBB2_2
; CHECK-NEXT:  .LBB2_3:
; CHECK-NEXT:    exrl %r1, .Ltmp1
; CHECK-NEXT:  .LBB2_4:
; CHECK-NEXT:    cgije %r1, -1, .LBB2_8
; CHECK-NEXT:  # %bb.5:
; CHECK-NEXT:    srlg %r0, %r1, 8
; CHECK-NEXT:    lgr %r3, %r2
; CHECK-NEXT:    cgije %r0, 0, .LBB2_7
; CHECK-NEXT:  .LBB2_6: # =>This Inner Loop Header: Depth=1
; CHECK-NEXT:    xc 0(256,%r3), 0(%r3)
; CHECK-NEXT:    la %r3, 256(%r3)
; CHECK-NEXT:    brctg %r0, .LBB2_6
; CHECK-NEXT:  .LBB2_7:
; CHECK-NEXT:    exrl %r1, .Ltmp1
; CHECK-NEXT:  .LBB2_8:
; CHECK-NEXT:    cgibe %r1, -1, 0(%r14)
; CHECK-NEXT:  .LBB2_9:
; CHECK-NEXT:    srlg %r0, %r1, 8
; CHECK-NEXT:    cgije %r0, 0, .LBB2_11
; CHECK-NEXT:  .LBB2_10: # =>This Inner Loop Header: Depth=1
; CHECK-NEXT:    xc 0(256,%r2), 0(%r2)
; CHECK-NEXT:    la %r2, 256(%r2)
; CHECK-NEXT:    brctg %r0, .LBB2_10
; CHECK-NEXT:  .LBB2_11:
; CHECK-NEXT:    exrl %r1, .Ltmp0
; CHECK-NEXT:    br %r14
  tail call void @llvm.memset.p0i8.i32(i8* %Addr, i8 0, i32 %Len, i1 false)
  tail call void @llvm.memset.p0i8.i32(i8* %Addr, i8 0, i32 %Len, i1 false)
  tail call void @llvm.memset.p0i8.i32(i8* %Addr, i8 0, i32 %Len, i1 false)
  ret void
}

; Test that a memset to nullptr compiles.
define void @fun3(i64 %Len) {
; CHECK-LABEL: fun3:
; CHECK:       # %bb.0:
; CHECK-NEXT:    aghi %r2, -1
; CHECK-NEXT:    cgibe %r2, -1, 0(%r14)
; CHECK-NEXT:  .LBB3_1:
; CHECK-NEXT:    srlg %r0, %r2, 8
; CHECK-NEXT:    lghi %r1, 0
; CHECK-NEXT:    cgije %r0, 0, .LBB3_3
; CHECK-NEXT:  .LBB3_2: # =>This Inner Loop Header: Depth=1
; CHECK-NEXT:    xc 0(256,%r1), 0(%r1)
; CHECK-NEXT:    la %r1, 256(%r1)
; CHECK-NEXT:    brctg %r0, .LBB3_2
; CHECK-NEXT:  .LBB3_3:
; CHECK-NEXT:    exrl %r2, .Ltmp2
; CHECK-NEXT:    br %r14
  call void @llvm.memset.p0i8.i64(i8* null, i8 0, i64 %Len, i1 false)
  ret void
}

; Test that a memset with a length argument that DAGCombiner will convert
; into a constant get the correct number of bytes set.
@Data = external hidden constant [1024 x i8], align 2
define void @fun4() {
; CHECK-LABEL: fun4:
; CHECK:       # %bb.0:
; CHECK-NEXT:    larl %r1, Data
; CHECK-NEXT:    xc 35(256,%r1), 35(%r1)
; CHECK-NEXT:    xc 291(256,%r1), 291(%r1)
; CHECK-NEXT:    xc 547(256,%r1), 547(%r1)
; CHECK-NEXT:    xc 803(221,%r1), 803(%r1)
; CHECK-NEXT:    mvghi 0(%r1), 989
; CHECK-NEXT:    br %r14
  call void @llvm.memset.p0i8.i64(
     i8* getelementptr inbounds ([1024 x i8], [1024 x i8]* @Data, i64 0, i64 35),
     i8 0,
     i64 sub (i64 add (i64 ptrtoint (i8* getelementptr inbounds ([1024 x i8],
                                     [1024 x i8]* @Data, i64 1, i64 0) to i64), i64 1),
              i64 add (i64 ptrtoint (i8* getelementptr inbounds ([1024 x i8],
                                     [1024 x i8]* @Data, i64 0, i64 35) to i64), i64 1)),
     i1 false)
  %i11 = getelementptr i8, i8* null,
     i64 sub (i64 add (i64 ptrtoint (i8* getelementptr inbounds ([1024 x i8],
                                     [1024 x i8]* @Data, i64 1, i64 0) to i64), i64 1),
              i64 add (i64 ptrtoint (i8* getelementptr inbounds ([1024 x i8],
                                     [1024 x i8]* @Data, i64 0, i64 35) to i64), i64 1))
  store i8* %i11, i8** undef, align 8
  ret void
}

; The same, with a resulting constant length of 0.
define void @fun5() {
; CHECK-LABEL: fun5:
; CHECK:       # %bb.0:
; CHECK-NEXT:    mvghi 0(%r1), 0
; CHECK-NEXT:    br %r14
  call void @llvm.memset.p0i8.i64(
     i8* getelementptr inbounds ([1024 x i8], [1024 x i8]* @Data, i64 0, i64 35),
     i8 0,
     i64 sub (i64 add (i64 ptrtoint (i8* getelementptr inbounds ([1024 x i8],
                                     [1024 x i8]* @Data, i64 1, i64 35) to i64), i64 1),
              i64 add (i64 ptrtoint (i8* getelementptr inbounds ([1024 x i8],
                                     [1024 x i8]* @Data, i64 1, i64 35) to i64), i64 1)),
     i1 false)
  %i11 = getelementptr i8, i8* null,
     i64 sub (i64 add (i64 ptrtoint (i8* getelementptr inbounds ([1024 x i8],
                                     [1024 x i8]* @Data, i64 1, i64 35) to i64), i64 1),
              i64 add (i64 ptrtoint (i8* getelementptr inbounds ([1024 x i8],
                                     [1024 x i8]* @Data, i64 1, i64 35) to i64), i64 1))
  store i8* %i11, i8** undef, align 8
  ret void
}

; The same, with a resulting constant length of 1.
define void @fun6() {
; CHECK-LABEL: fun6:
; CHECK:       # %bb.0:
; CHECK-NEXT:    larl %r1, Data
; CHECK-NEXT:    xc 35(1,%r1), 35(%r1)
; CHECK-NEXT:    mvghi 0(%r1), 1
; CHECK-NEXT:    br %r14
  call void @llvm.memset.p0i8.i64(
     i8* getelementptr inbounds ([1024 x i8], [1024 x i8]* @Data, i64 0, i64 35),
     i8 0,
     i64 sub (i64 add (i64 ptrtoint (i8* getelementptr inbounds ([1024 x i8],
                                     [1024 x i8]* @Data, i64 1, i64 36) to i64), i64 1),
              i64 add (i64 ptrtoint (i8* getelementptr inbounds ([1024 x i8],
                                     [1024 x i8]* @Data, i64 1, i64 35) to i64), i64 1)),
     i1 false)
  %i11 = getelementptr i8, i8* null,
     i64 sub (i64 add (i64 ptrtoint (i8* getelementptr inbounds ([1024 x i8],
                                     [1024 x i8]* @Data, i64 1, i64 36) to i64), i64 1),
              i64 add (i64 ptrtoint (i8* getelementptr inbounds ([1024 x i8],
                                     [1024 x i8]* @Data, i64 1, i64 35) to i64), i64 1))
  store i8* %i11, i8** undef, align 8
  ret void
}

; The same, with a resulting constant length of 256.
define void @fun7() {
; CHECK-LABEL: fun7:
; CHECK:       # %bb.0:
; CHECK-NEXT:    larl %r1, Data
; CHECK-NEXT:    xc 35(256,%r1), 35(%r1)
; CHECK-NEXT:    mvghi 0(%r1), 256
; CHECK-NEXT:    br %r14
  call void @llvm.memset.p0i8.i64(
     i8* getelementptr inbounds ([1024 x i8], [1024 x i8]* @Data, i64 0, i64 35),
     i8 0,
     i64 sub (i64 add (i64 ptrtoint (i8* getelementptr inbounds ([1024 x i8],
                                     [1024 x i8]* @Data, i64 1, i64 291) to i64), i64 1),
              i64 add (i64 ptrtoint (i8* getelementptr inbounds ([1024 x i8],
                                     [1024 x i8]* @Data, i64 1, i64 35) to i64), i64 1)),
     i1 false)
  %i11 = getelementptr i8, i8* null,
     i64 sub (i64 add (i64 ptrtoint (i8* getelementptr inbounds ([1024 x i8],
                                     [1024 x i8]* @Data, i64 1, i64 291) to i64), i64 1),
              i64 add (i64 ptrtoint (i8* getelementptr inbounds ([1024 x i8],
                                     [1024 x i8]* @Data, i64 1, i64 35) to i64), i64 1))
  store i8* %i11, i8** undef, align 8
  ret void
}

; The same, with a resulting constant length of 257.
define void @fun8() {
; CHECK-LABEL: fun8:
; CHECK:       # %bb.0:
; CHECK-NEXT:    larl %r1, Data
; CHECK-NEXT:    xc 35(256,%r1), 35(%r1)
; CHECK-NEXT:    xc 291(1,%r1), 291(%r1)
; CHECK-NEXT:    mvghi 0(%r1), 257
; CHECK-NEXT:    br %r14
  call void @llvm.memset.p0i8.i64(
     i8* getelementptr inbounds ([1024 x i8], [1024 x i8]* @Data, i64 0, i64 35),
     i8 0,
     i64 sub (i64 add (i64 ptrtoint (i8* getelementptr inbounds ([1024 x i8],
                                     [1024 x i8]* @Data, i64 1, i64 292) to i64), i64 1),
              i64 add (i64 ptrtoint (i8* getelementptr inbounds ([1024 x i8],
                                     [1024 x i8]* @Data, i64 1, i64 35) to i64), i64 1)),
     i1 false)
  %i11 = getelementptr i8, i8* null,
     i64 sub (i64 add (i64 ptrtoint (i8* getelementptr inbounds ([1024 x i8],
                                     [1024 x i8]* @Data, i64 1, i64 292) to i64), i64 1),
              i64 add (i64 ptrtoint (i8* getelementptr inbounds ([1024 x i8],
                                     [1024 x i8]* @Data, i64 1, i64 35) to i64), i64 1))
  store i8* %i11, i8** undef, align 8
  ret void
}

; CHECK:       .Ltmp2:
; CHECK-NEXT: 	 xc 0(1,%r1), 0(%r1)
; CHECK-NEXT:  .Ltmp0:
; CHECK-NEXT:    xc 0(1,%r2), 0(%r2)
; CHECK-NEXT:  .Ltmp1:
; CHECK-NEXT:    xc 0(1,%r3), 0(%r3)

declare void @llvm.memset.p0i8.i64(i8* nocapture writeonly, i8, i64, i1 immarg)
declare void @llvm.memset.p0i8.i32(i8* nocapture writeonly, i8, i32, i1 immarg)