test/CodeGen/ARM/fp16-load-store.ll

; RUN: llc < %s -mtriple armv8a--none-eabi -mattr=+fullfp16 | FileCheck %s

define void @load_zero(ptr %in, ptr %out) {
entry:
; CHECK-LABEL: load_zero:
; CHECK: vldr.16 {{s[0-9]+}}, [r0]
  %load = load half, ptr %in, align 2
  store half %load, ptr %out
  ret void
}

define void @load_255(ptr %in, ptr %out) {
entry:
; CHECK-LABEL: load_255:
; CHECK: vldr.16 {{s[0-9]+}}, [r0, #510]
  %arrayidx = getelementptr inbounds half, ptr %in, i32 255
  %load = load half, ptr %arrayidx, align 2
  store half %load, ptr %out
  ret void
}

define void @load_256(ptr %in, ptr %out) {
entry:
; CHECK-LABEL: load_256:
; CHECK: add     [[ADDR:r[0-9]+]], r0, #512
; CHECK: vldr.16 {{s[0-9]+}}, [[[ADDR]]]
  %arrayidx = getelementptr inbounds half, ptr %in, i32 256
  %load = load half, ptr %arrayidx, align 2
  store half %load, ptr %out
  ret void
}

define void @load_neg_255(ptr %in, ptr %out) {
entry:
; CHECK-LABEL: load_neg_255:
; CHECK: vldr.16 {{s[0-9]+}}, [r0, #-510]
  %arrayidx = getelementptr inbounds half, ptr %in, i32 -255
  %load = load half, ptr %arrayidx, align 2
  store half %load, ptr %out
  ret void
}

define void @load_neg_256(ptr %in, ptr %out) {
entry:
; CHECK-LABEL: load_neg_256:
; CHECK: sub     [[ADDR:r[0-9]+]], r0, #512
; CHECK: vldr.16 {{s[0-9]+}}, [[[ADDR]]]
  %arrayidx = getelementptr inbounds half, ptr %in, i32 -256
  %load = load half, ptr %arrayidx, align 2
  store half %load, ptr %out
  ret void
}

define void @store_zero(ptr %in, ptr %out) {
entry:
; CHECK-LABEL: store_zero:
  %load = load half, ptr %in, align 2
; CHECK: vstr.16 {{s[0-9]+}}, [r1]
  store half %load, ptr %out
  ret void
}

define void @store_255(ptr %in, ptr %out) {
entry:
; CHECK-LABEL: store_255:
  %load = load half, ptr %in, align 2
; CHECK: vstr.16 {{s[0-9]+}}, [r1, #510]
  %arrayidx = getelementptr inbounds half, ptr %out, i32 255
  store half %load, ptr %arrayidx
  ret void
}

define void @store_256(ptr %in, ptr %out) {
entry:
; CHECK-LABEL: store_256:
  %load = load half, ptr %in, align 2
; CHECK: add     [[ADDR:r[0-9]+]], r1, #512
; CHECK: vstr.16 {{s[0-9]+}}, [[[ADDR]]]
  %arrayidx = getelementptr inbounds half, ptr %out, i32 256
  store half %load, ptr %arrayidx
  ret void
}

define void @store_neg_255(ptr %in, ptr %out) {
entry:
; CHECK-LABEL: store_neg_255:
  %load = load half, ptr %in, align 2
; CHECK: vstr.16 {{s[0-9]+}}, [r1, #-510]
  %arrayidx = getelementptr inbounds half, ptr %out, i32 -255
  store half %load, ptr %arrayidx
  ret void
}

define void @store_neg_256(ptr %in, ptr %out) {
entry:
; CHECK-LABEL: store_neg_256:
  %load = load half, ptr %in, align 2
; CHECK: sub     [[ADDR:r[0-9]+]], r1, #512
; CHECK: vstr.16 {{s[0-9]+}}, [[[ADDR]]]
  %arrayidx = getelementptr inbounds half, ptr %out, i32 -256
  store half %load, ptr %arrayidx
  ret void
}