llvm/test/CodeGen/AArch64/arm64-indexed-memory.ll

; RUN: llc < %s -mtriple=arm64-eabi -aarch64-redzone | FileCheck %s

define i64* @store64(i64* %ptr, i64 %index, i64 %spacing) {
; CHECK-LABEL: store64:
; CHECK: str x{{[0-9+]}}, [x{{[0-9+]}}], #8
; CHECK: ret
  %incdec.ptr = getelementptr inbounds i64, i64* %ptr, i64 1
  store i64 %spacing, i64* %ptr, align 4
  ret i64* %incdec.ptr
}

define i64* @store64idxpos256(i64* %ptr, i64 %index, i64 %spacing) {
; CHECK-LABEL: store64idxpos256:
; CHECK: add x{{[0-9+]}}, x{{[0-9+]}}, #256
; CHECK: str x{{[0-9+]}}, [x{{[0-9+]}}]
; CHECK: ret
  %incdec.ptr = getelementptr inbounds i64, i64* %ptr, i64 32
  store i64 %spacing, i64* %ptr, align 4
  ret i64* %incdec.ptr
}

define i64* @store64idxneg256(i64* %ptr, i64 %index, i64 %spacing) {
; CHECK-LABEL: store64idxneg256:
; CHECK: str x{{[0-9+]}}, [x{{[0-9+]}}], #-256
; CHECK: ret
  %incdec.ptr = getelementptr inbounds i64, i64* %ptr, i64 -32
  store i64 %spacing, i64* %ptr, align 4
  ret i64* %incdec.ptr
}

define i32* @store32(i32* %ptr, i32 %index, i32 %spacing) {
; CHECK-LABEL: store32:
; CHECK: str w{{[0-9+]}}, [x{{[0-9+]}}], #4
; CHECK: ret
  %incdec.ptr = getelementptr inbounds i32, i32* %ptr, i64 1
  store i32 %spacing, i32* %ptr, align 4
  ret i32* %incdec.ptr
}

define i32* @store32idxpos256(i32* %ptr, i32 %index, i32 %spacing) {
; CHECK-LABEL: store32idxpos256:
; CHECK: add x{{[0-9+]}}, x{{[0-9+]}}, #256
; CHECK: str w{{[0-9+]}}, [x{{[0-9+]}}]
; CHECK: ret
  %incdec.ptr = getelementptr inbounds i32, i32* %ptr, i64 64
  store i32 %spacing, i32* %ptr, align 4
  ret i32* %incdec.ptr
}

define i32* @store32idxneg256(i32* %ptr, i32 %index, i32 %spacing) {
; CHECK-LABEL: store32idxneg256:
; CHECK: str w{{[0-9+]}}, [x{{[0-9+]}}], #-256
; CHECK: ret
  %incdec.ptr = getelementptr inbounds i32, i32* %ptr, i64 -64
  store i32 %spacing, i32* %ptr, align 4
  ret i32* %incdec.ptr
}

define i16* @store16(i16* %ptr, i16 %index, i16 %spacing) {
; CHECK-LABEL: store16:
; CHECK: strh w{{[0-9+]}}, [x{{[0-9+]}}], #2
; CHECK: ret
  %incdec.ptr = getelementptr inbounds i16, i16* %ptr, i64 1
  store i16 %spacing, i16* %ptr, align 4
  ret i16* %incdec.ptr
}

define i16* @store16idxpos256(i16* %ptr, i16 %index, i16 %spacing) {
; CHECK-LABEL: store16idxpos256:
; CHECK: add x{{[0-9+]}}, x{{[0-9+]}}, #256
; CHECK: strh w{{[0-9+]}}, [x{{[0-9+]}}]
; CHECK: ret
  %incdec.ptr = getelementptr inbounds i16, i16* %ptr, i64 128
  store i16 %spacing, i16* %ptr, align 4
  ret i16* %incdec.ptr
}

define i16* @store16idxneg256(i16* %ptr, i16 %index, i16 %spacing) {
; CHECK-LABEL: store16idxneg256:
; CHECK: strh w{{[0-9+]}}, [x{{[0-9+]}}], #-256
; CHECK: ret
  %incdec.ptr = getelementptr inbounds i16, i16* %ptr, i64 -128
  store i16 %spacing, i16* %ptr, align 4
  ret i16* %incdec.ptr
}

define i8* @store8(i8* %ptr, i8 %index, i8 %spacing) {
; CHECK-LABEL: store8:
; CHECK: strb w{{[0-9+]}}, [x{{[0-9+]}}], #1
; CHECK: ret
  %incdec.ptr = getelementptr inbounds i8, i8* %ptr, i64 1
  store i8 %spacing, i8* %ptr, align 4
  ret i8* %incdec.ptr
}

define i8* @store8idxpos256(i8* %ptr, i8 %index, i8 %spacing) {
; CHECK-LABEL: store8idxpos256:
; CHECK: add x{{[0-9+]}}, x{{[0-9+]}}, #256
; CHECK: strb w{{[0-9+]}}, [x{{[0-9+]}}]
; CHECK: ret
  %incdec.ptr = getelementptr inbounds i8, i8* %ptr, i64 256
  store i8 %spacing, i8* %ptr, align 4
  ret i8* %incdec.ptr
}

define i8* @store8idxneg256(i8* %ptr, i8 %index, i8 %spacing) {
; CHECK-LABEL: store8idxneg256:
; CHECK: strb w{{[0-9+]}}, [x{{[0-9+]}}], #-256
; CHECK: ret
  %incdec.ptr = getelementptr inbounds i8, i8* %ptr, i64 -256
  store i8 %spacing, i8* %ptr, align 4
  ret i8* %incdec.ptr
}

define i32* @truncst64to32(i32* %ptr, i32 %index, i64 %spacing) {
; CHECK-LABEL: truncst64to32:
; CHECK: str w{{[0-9+]}}, [x{{[0-9+]}}], #4
; CHECK: ret
  %incdec.ptr = getelementptr inbounds i32, i32* %ptr, i64 1
  %trunc = trunc i64 %spacing to i32
  store i32 %trunc, i32* %ptr, align 4
  ret i32* %incdec.ptr
}

define i16* @truncst64to16(i16* %ptr, i16 %index, i64 %spacing) {
; CHECK-LABEL: truncst64to16:
; CHECK: strh w{{[0-9+]}}, [x{{[0-9+]}}], #2
; CHECK: ret
  %incdec.ptr = getelementptr inbounds i16, i16* %ptr, i64 1
  %trunc = trunc i64 %spacing to i16
  store i16 %trunc, i16* %ptr, align 4
  ret i16* %incdec.ptr
}

define i8* @truncst64to8(i8* %ptr, i8 %index, i64 %spacing) {
; CHECK-LABEL: truncst64to8:
; CHECK: strb w{{[0-9+]}}, [x{{[0-9+]}}], #1
; CHECK: ret
  %incdec.ptr = getelementptr inbounds i8, i8* %ptr, i64 1
  %trunc = trunc i64 %spacing to i8
  store i8 %trunc, i8* %ptr, align 4
  ret i8* %incdec.ptr
}


define half* @storef16(half* %ptr, half %index, half %spacing) nounwind {
; CHECK-LABEL: storef16:
; CHECK: str h{{[0-9+]}}, [x{{[0-9+]}}], #2
; CHECK: ret
  %incdec.ptr = getelementptr inbounds half, half* %ptr, i64 1
  store half %spacing, half* %ptr, align 2
  ret half* %incdec.ptr
}

define float* @storef32(float* %ptr, float %index, float %spacing) {
; CHECK-LABEL: storef32:
; CHECK: str s{{[0-9+]}}, [x{{[0-9+]}}], #4
; CHECK: ret
  %incdec.ptr = getelementptr inbounds float, float* %ptr, i64 1
  store float %spacing, float* %ptr, align 4
  ret float* %incdec.ptr
}

define double* @storef64(double* %ptr, double %index, double %spacing) {
; CHECK-LABEL: storef64:
; CHECK: str d{{[0-9+]}}, [x{{[0-9+]}}], #8
; CHECK: ret
  %incdec.ptr = getelementptr inbounds double, double* %ptr, i64 1
  store double %spacing, double* %ptr, align 4
  ret double* %incdec.ptr
}


define double* @pref64(double* %ptr, double %spacing) {
; CHECK-LABEL: pref64:
; CHECK:      str d0, [x0, #32]!
; CHECK-NEXT: ret
  %incdec.ptr = getelementptr inbounds double, double* %ptr, i64 4
  store double %spacing, double* %incdec.ptr, align 4
  ret double *%incdec.ptr
}

define float* @pref32(float* %ptr, float %spacing) {
; CHECK-LABEL: pref32:
; CHECK:      str s0, [x0, #12]!
; CHECK-NEXT: ret
  %incdec.ptr = getelementptr inbounds float, float* %ptr, i64 3
  store float %spacing, float* %incdec.ptr, align 4
  ret float *%incdec.ptr
}

define half* @pref16(half* %ptr, half %spacing) nounwind {
; CHECK-LABEL: pref16:
; CHECK:      str h0, [x0, #6]!
; CHECK-NEXT: ret
  %incdec.ptr = getelementptr inbounds half, half* %ptr, i64 3
  store half %spacing, half* %incdec.ptr, align 2
  ret half *%incdec.ptr
}

define i64* @pre64(i64* %ptr, i64 %spacing) {
; CHECK-LABEL: pre64:
; CHECK:      str x1, [x0, #16]!
; CHECK-NEXT: ret
  %incdec.ptr = getelementptr inbounds i64, i64* %ptr, i64 2
  store i64 %spacing, i64* %incdec.ptr, align 4
  ret i64 *%incdec.ptr
}

define i64* @pre64idxpos256(i64* %ptr, i64 %spacing) {
; CHECK-LABEL: pre64idxpos256:
; CHECK:      add x8, x0, #256
; CHECK-NEXT: str x1, [x0, #256]
; CHECK-NEXT: mov x0, x8
; CHECK-NEXT: ret
  %incdec.ptr = getelementptr inbounds i64, i64* %ptr, i64 32
  store i64 %spacing, i64* %incdec.ptr, align 4
  ret i64 *%incdec.ptr
}

define i64* @pre64idxneg256(i64* %ptr, i64 %spacing) {
; CHECK-LABEL: pre64idxneg256:
; CHECK:      str x1, [x0, #-256]!
; CHECK-NEXT: ret
  %incdec.ptr = getelementptr inbounds i64, i64* %ptr, i64 -32
  store i64 %spacing, i64* %incdec.ptr, align 4
  ret i64 *%incdec.ptr
}

define i32* @pre32(i32* %ptr, i32 %spacing) {
; CHECK-LABEL: pre32:
; CHECK:      str w1, [x0, #8]!
; CHECK-NEXT: ret
  %incdec.ptr = getelementptr inbounds i32, i32* %ptr, i64 2
  store i32 %spacing, i32* %incdec.ptr, align 4
  ret i32 *%incdec.ptr
}

define i32* @pre32idxpos256(i32* %ptr, i32 %spacing) {
; CHECK-LABEL: pre32idxpos256:
; CHECK:      add x8, x0, #256
; CHECK-NEXT: str w1, [x0, #256]
; CHECK-NEXT: mov x0, x8
; CHECK-NEXT: ret
  %incdec.ptr = getelementptr inbounds i32, i32* %ptr, i64 64
  store i32 %spacing, i32* %incdec.ptr, align 4
  ret i32 *%incdec.ptr
}

define i32* @pre32idxneg256(i32* %ptr, i32 %spacing) {
; CHECK-LABEL: pre32idxneg256:
; CHECK:      str w1, [x0, #-256]!
; CHECK-NEXT: ret
  %incdec.ptr = getelementptr inbounds i32, i32* %ptr, i64 -64
  store i32 %spacing, i32* %incdec.ptr, align 4
  ret i32 *%incdec.ptr
}

define i16* @pre16(i16* %ptr, i16 %spacing) {
; CHECK-LABEL: pre16:
; CHECK:      strh w1, [x0, #4]!
; CHECK-NEXT: ret
  %incdec.ptr = getelementptr inbounds i16, i16* %ptr, i64 2
  store i16 %spacing, i16* %incdec.ptr, align 4
  ret i16 *%incdec.ptr
}

define i16* @pre16idxpos256(i16* %ptr, i16 %spacing) {
; CHECK-LABEL: pre16idxpos256:
; CHECK:      add x8, x0, #256
; CHECK-NEXT: strh w1, [x0, #256]
; CHECK-NEXT: mov x0, x8
; CHECK-NEXT: ret
  %incdec.ptr = getelementptr inbounds i16, i16* %ptr, i64 128
  store i16 %spacing, i16* %incdec.ptr, align 4
  ret i16 *%incdec.ptr
}

define i16* @pre16idxneg256(i16* %ptr, i16 %spacing) {
; CHECK-LABEL: pre16idxneg256:
; CHECK:      strh w1, [x0, #-256]!
; CHECK-NEXT: ret
  %incdec.ptr = getelementptr inbounds i16, i16* %ptr, i64 -128
  store i16 %spacing, i16* %incdec.ptr, align 4
  ret i16 *%incdec.ptr
}

define i8* @pre8(i8* %ptr, i8 %spacing) {
; CHECK-LABEL: pre8:
; CHECK:      strb w1, [x0, #2]!
; CHECK-NEXT: ret
  %incdec.ptr = getelementptr inbounds i8, i8* %ptr, i64 2
  store i8 %spacing, i8* %incdec.ptr, align 4
  ret i8 *%incdec.ptr
}

define i8* @pre8idxpos256(i8* %ptr, i8 %spacing) {
; CHECK-LABEL: pre8idxpos256:
; CHECK:      add x8, x0, #256
; CHECK-NEXT: strb w1, [x0, #256]
; CHECK-NEXT: mov x0, x8
; CHECK-NEXT: ret
  %incdec.ptr = getelementptr inbounds i8, i8* %ptr, i64 256
  store i8 %spacing, i8* %incdec.ptr, align 4
  ret i8 *%incdec.ptr
}

define i8* @pre8idxneg256(i8* %ptr, i8 %spacing) {
; CHECK-LABEL: pre8idxneg256:
; CHECK:      strb w1, [x0, #-256]!
; CHECK-NEXT: ret
  %incdec.ptr = getelementptr inbounds i8, i8* %ptr, i64 -256
  store i8 %spacing, i8* %incdec.ptr, align 4
  ret i8 *%incdec.ptr
}

define i32* @pretrunc64to32(i32* %ptr, i64 %spacing) {
; CHECK-LABEL: pretrunc64to32:
; CHECK:      str w1, [x0, #8]!
; CHECK-NEXT: ret
  %incdec.ptr = getelementptr inbounds i32, i32* %ptr, i64 2
  %trunc = trunc i64 %spacing to i32
  store i32 %trunc, i32* %incdec.ptr, align 4
  ret i32 *%incdec.ptr
}

define i16* @pretrunc64to16(i16* %ptr, i64 %spacing) {
; CHECK-LABEL: pretrunc64to16:
; CHECK:      strh w1, [x0, #4]!
; CHECK-NEXT: ret
  %incdec.ptr = getelementptr inbounds i16, i16* %ptr, i64 2
  %trunc = trunc i64 %spacing to i16
  store i16 %trunc, i16* %incdec.ptr, align 4
  ret i16 *%incdec.ptr
}

define i8* @pretrunc64to8(i8* %ptr, i64 %spacing) {
; CHECK-LABEL: pretrunc64to8:
; CHECK:      strb w1, [x0, #2]!
; CHECK-NEXT: ret
  %incdec.ptr = getelementptr inbounds i8, i8* %ptr, i64 2
  %trunc = trunc i64 %spacing to i8
  store i8 %trunc, i8* %incdec.ptr, align 4
  ret i8 *%incdec.ptr
}

;-----
; Pre-indexed loads
;-----
define double* @preidxf64(double* %src, double* %out) {
; CHECK-LABEL: preidxf64:
; CHECK: ldr     d0, [x0, #8]!
; CHECK: str     d0, [x1]
; CHECK: ret
  %ptr = getelementptr inbounds double, double* %src, i64 1
  %tmp = load double, double* %ptr, align 4
  store double %tmp, double* %out, align 4
  ret double* %ptr
}

define float* @preidxf32(float* %src, float* %out) {
; CHECK-LABEL: preidxf32:
; CHECK: ldr     s0, [x0, #4]!
; CHECK: str     s0, [x1]
; CHECK: ret
  %ptr = getelementptr inbounds float, float* %src, i64 1
  %tmp = load float, float* %ptr, align 4
  store float %tmp, float* %out, align 4
  ret float* %ptr
}

define half* @preidxf16(half* %src, half* %out) {
; CHECK-LABEL: preidxf16:
; CHECK: ldr     h0, [x0, #2]!
; CHECK: str     h0, [x1]
; CHECK: ret
  %ptr = getelementptr inbounds half, half* %src, i64 1
  %tmp = load half, half* %ptr, align 2
  store half %tmp, half* %out, align 2
  ret half* %ptr
}

define i64* @preidx64(i64* %src, i64* %out) {
; CHECK-LABEL: preidx64:
; CHECK: ldr     x[[REG:[0-9]+]], [x0, #8]!
; CHECK: str     x[[REG]], [x1]
; CHECK: ret
  %ptr = getelementptr inbounds i64, i64* %src, i64 1
  %tmp = load i64, i64* %ptr, align 4
  store i64 %tmp, i64* %out, align 4
  ret i64* %ptr
}

define i32* @preidx32(i32* %src, i32* %out) {
; CHECK: ldr     w[[REG:[0-9]+]], [x0, #4]!
; CHECK: str     w[[REG]], [x1]
; CHECK: ret
  %ptr = getelementptr inbounds i32, i32* %src, i64 1
  %tmp = load i32, i32* %ptr, align 4
  store i32 %tmp, i32* %out, align 4
  ret i32* %ptr
}

define i16* @preidx16zext32(i16* %src, i32* %out) {
; CHECK: ldrh    w[[REG:[0-9]+]], [x0, #2]!
; CHECK: str     w[[REG]], [x1]
; CHECK: ret
  %ptr = getelementptr inbounds i16, i16* %src, i64 1
  %tmp = load i16, i16* %ptr, align 4
  %ext = zext i16 %tmp to i32
  store i32 %ext, i32* %out, align 4
  ret i16* %ptr
}

define i16* @preidx16zext64(i16* %src, i64* %out) {
; CHECK: ldrh    w[[REG:[0-9]+]], [x0, #2]!
; CHECK: str     x[[REG]], [x1]
; CHECK: ret
  %ptr = getelementptr inbounds i16, i16* %src, i64 1
  %tmp = load i16, i16* %ptr, align 4
  %ext = zext i16 %tmp to i64
  store i64 %ext, i64* %out, align 4
  ret i16* %ptr
}

define i8* @preidx8zext32(i8* %src, i32* %out) {
; CHECK: ldrb    w[[REG:[0-9]+]], [x0, #1]!
; CHECK: str     w[[REG]], [x1]
; CHECK: ret
  %ptr = getelementptr inbounds i8, i8* %src, i64 1
  %tmp = load i8, i8* %ptr, align 4
  %ext = zext i8 %tmp to i32
  store i32 %ext, i32* %out, align 4
  ret i8* %ptr
}

define i8* @preidx8zext64(i8* %src, i64* %out) {
; CHECK: ldrb    w[[REG:[0-9]+]], [x0, #1]!
; CHECK: str     x[[REG]], [x1]
; CHECK: ret
  %ptr = getelementptr inbounds i8, i8* %src, i64 1
  %tmp = load i8, i8* %ptr, align 4
  %ext = zext i8 %tmp to i64
  store i64 %ext, i64* %out, align 4
  ret i8* %ptr
}

define i32* @preidx32sext64(i32* %src, i64* %out) {
; CHECK: ldrsw   x[[REG:[0-9]+]], [x0, #4]!
; CHECK: str     x[[REG]], [x1]
; CHECK: ret
  %ptr = getelementptr inbounds i32, i32* %src, i64 1
  %tmp = load i32, i32* %ptr, align 4
  %ext = sext i32 %tmp to i64
  store i64 %ext, i64* %out, align 8
  ret i32* %ptr
}

define i16* @preidx16sext32(i16* %src, i32* %out) {
; CHECK: ldrsh   w[[REG:[0-9]+]], [x0, #2]!
; CHECK: str     w[[REG]], [x1]
; CHECK: ret
  %ptr = getelementptr inbounds i16, i16* %src, i64 1
  %tmp = load i16, i16* %ptr, align 4
  %ext = sext i16 %tmp to i32
  store i32 %ext, i32* %out, align 4
  ret i16* %ptr
}

define i16* @preidx16sext64(i16* %src, i64* %out) {
; CHECK: ldrsh   x[[REG:[0-9]+]], [x0, #2]!
; CHECK: str     x[[REG]], [x1]
; CHECK: ret
  %ptr = getelementptr inbounds i16, i16* %src, i64 1
  %tmp = load i16, i16* %ptr, align 4
  %ext = sext i16 %tmp to i64
  store i64 %ext, i64* %out, align 4
  ret i16* %ptr
}

define i8* @preidx8sext32(i8* %src, i32* %out) {
; CHECK: ldrsb   w[[REG:[0-9]+]], [x0, #1]!
; CHECK: str     w[[REG]], [x1]
; CHECK: ret
  %ptr = getelementptr inbounds i8, i8* %src, i64 1
  %tmp = load i8, i8* %ptr, align 4
  %ext = sext i8 %tmp to i32
  store i32 %ext, i32* %out, align 4
  ret i8* %ptr
}

define i8* @preidx8sext64(i8* %src, i64* %out) {
; CHECK: ldrsb   x[[REG:[0-9]+]], [x0, #1]!
; CHECK: str     x[[REG]], [x1]
; CHECK: ret
  %ptr = getelementptr inbounds i8, i8* %src, i64 1
  %tmp = load i8, i8* %ptr, align 4
  %ext = sext i8 %tmp to i64
  store i64 %ext, i64* %out, align 4
  ret i8* %ptr
}

; This test checks if illegal post-index is generated

define i64* @postidx_clobber(i64* %addr) nounwind noinline ssp {
; CHECK-LABEL: postidx_clobber:
; CHECK-NOT: str     x0, [x0], #8
; ret
 %paddr = bitcast i64* %addr to i64**
 store i64* %addr, i64** %paddr
 %newaddr = getelementptr i64, i64* %addr, i32 1
 ret i64* %newaddr
}