llvm/llvm/test/CodeGen/NVPTX/st-addrspace.ll

; RUN: llc < %s -march=nvptx -mcpu=sm_20 | FileCheck %s --check-prefixes=ALL,G32,LS32
; RUN: llc < %s -march=nvptx64 -mcpu=sm_20 | FileCheck %s --check-prefixes=ALL,G64,LS64
; RUN: llc < %s -march=nvptx64 -mcpu=sm_20 --nvptx-short-ptr | FileCheck %s --check-prefixes=G64,LS32
; RUN: %if ptxas && !ptxas-12.0 %{ llc < %s -march=nvptx -mcpu=sm_20 | %ptxas-verify %}
; RUN: %if ptxas %{ llc < %s -march=nvptx64 -mcpu=sm_20 | %ptxas-verify %}
; RUN: %if ptxas %{ llc < %s -march=nvptx64 -mcpu=sm_20 --nvptx-short-ptr | %ptxas-verify %}

;; i8
; ALL-LABEL: st_global_i8
define void @st_global_i8(ptr addrspace(1) %ptr, i8 %a) {
; G32: st.global.u8 [%r{{[0-9]+}}], %rs{{[0-9]+}}
; G64: st.global.u8 [%rd{{[0-9]+}}], %rs{{[0-9]+}}
; ALL: ret
  store i8 %a, ptr addrspace(1) %ptr
  ret void
}
; ALL-LABEL: st_shared_i8
define void @st_shared_i8(ptr addrspace(3) %ptr, i8 %a) {
; LS32: st.shared.u8 [%r{{[0-9]+}}], %rs{{[0-9]+}}
; LS64: st.shared.u8 [%rd{{[0-9]+}}], %rs{{[0-9]+}}
; ALL: ret
  store i8 %a, ptr addrspace(3) %ptr
  ret void
}
; ALL-LABEL: st_local_i8
define void @st_local_i8(ptr addrspace(5) %ptr, i8 %a) {
; LS32: st.local.u8 [%r{{[0-9]+}}], %rs{{[0-9]+}}
; LS64: st.local.u8 [%rd{{[0-9]+}}], %rs{{[0-9]+}}
; ALL: ret
  store i8 %a, ptr addrspace(5) %ptr
  ret void
}

;; i16
; ALL-LABEL: st_global_i16
define void @st_global_i16(ptr addrspace(1) %ptr, i16 %a) {
; G32: st.global.u16 [%r{{[0-9]+}}], %rs{{[0-9]+}}
; G64: st.global.u16 [%rd{{[0-9]+}}], %rs{{[0-9]+}}
; ALL: ret
  store i16 %a, ptr addrspace(1) %ptr
  ret void
}
; ALL-LABEL: st_shared_i16
define void @st_shared_i16(ptr addrspace(3) %ptr, i16 %a) {
; LS32: st.shared.u16 [%r{{[0-9]+}}], %rs{{[0-9]+}}
; LS64: st.shared.u16 [%rd{{[0-9]+}}], %rs{{[0-9]+}}
; ALL: ret
  store i16 %a, ptr addrspace(3) %ptr
  ret void
}
; ALL-LABEL: st_local_i16
define void @st_local_i16(ptr addrspace(5) %ptr, i16 %a) {
; LS32: st.local.u16 [%r{{[0-9]+}}], %rs{{[0-9]+}}
; LS64: st.local.u16 [%rd{{[0-9]+}}], %rs{{[0-9]+}}
; ALL: ret
  store i16 %a, ptr addrspace(5) %ptr
  ret void
}

;; i32
; ALL-LABEL: st_global_i32
define void @st_global_i32(ptr addrspace(1) %ptr, i32 %a) {
; G32: st.global.u32 [%r{{[0-9]+}}], %r{{[0-9]+}}
; G64: st.global.u32 [%rd{{[0-9]+}}], %r{{[0-9]+}}
; ALL: ret
  store i32 %a, ptr addrspace(1) %ptr
  ret void
}
; ALL-LABEL: st_shared_i32
define void @st_shared_i32(ptr addrspace(3) %ptr, i32 %a) {
; LS32: st.shared.u32 [%r{{[0-9]+}}], %r{{[0-9]+}}
; LS64: st.shared.u32 [%rd{{[0-9]+}}], %r{{[0-9]+}}
; PTX64: ret
  store i32 %a, ptr addrspace(3) %ptr
  ret void
}
; ALL-LABEL: st_local_i32
define void @st_local_i32(ptr addrspace(5) %ptr, i32 %a) {
; LS32: st.local.u32 [%r{{[0-9]+}}], %r{{[0-9]+}}
; LS64: st.local.u32 [%rd{{[0-9]+}}], %r{{[0-9]+}}
; ALL: ret
  store i32 %a, ptr addrspace(5) %ptr
  ret void
}

;; i64
; ALL-LABEL: st_global_i64
define void @st_global_i64(ptr addrspace(1) %ptr, i64 %a) {
; G32: st.global.u64 [%r{{[0-9]+}}], %rd{{[0-9]+}}
; G64: st.global.u64 [%rd{{[0-9]+}}], %rd{{[0-9]+}}
; ALL: ret
  store i64 %a, ptr addrspace(1) %ptr
  ret void
}
; ALL-LABEL: st_shared_i64
define void @st_shared_i64(ptr addrspace(3) %ptr, i64 %a) {
; LS32: st.shared.u64 [%r{{[0-9]+}}], %rd{{[0-9]+}}
; LS64: st.shared.u64 [%rd{{[0-9]+}}], %rd{{[0-9]+}}
; ALL: ret
  store i64 %a, ptr addrspace(3) %ptr
  ret void
}
; ALL-LABEL: st_local_i64
define void @st_local_i64(ptr addrspace(5) %ptr, i64 %a) {
; LS32: st.local.u64 [%r{{[0-9]+}}], %rd{{[0-9]+}}
; LS64: st.local.u64 [%rd{{[0-9]+}}], %rd{{[0-9]+}}
; ALL: ret
  store i64 %a, ptr addrspace(5) %ptr
  ret void
}

;; f32
; ALL-LABEL: st_global_f32
define void @st_global_f32(ptr addrspace(1) %ptr, float %a) {
; G32: st.global.f32 [%r{{[0-9]+}}], %f{{[0-9]+}}
; G64: st.global.f32 [%rd{{[0-9]+}}], %f{{[0-9]+}}
; ALL: ret
  store float %a, ptr addrspace(1) %ptr
  ret void
}
; ALL-LABEL: st_shared_f32
define void @st_shared_f32(ptr addrspace(3) %ptr, float %a) {
; LS32: st.shared.f32 [%r{{[0-9]+}}], %f{{[0-9]+}}
; LS64: st.shared.f32 [%rd{{[0-9]+}}], %f{{[0-9]+}}
; ALL: ret
  store float %a, ptr addrspace(3) %ptr
  ret void
}
; ALL-LABEL: st_local_f32
define void @st_local_f32(ptr addrspace(5) %ptr, float %a) {
; LS32: st.local.f32 [%r{{[0-9]+}}], %f{{[0-9]+}}
; LS64: st.local.f32 [%rd{{[0-9]+}}], %f{{[0-9]+}}
; ALL: ret
  store float %a, ptr addrspace(5) %ptr
  ret void
}

;; f64
; ALL-LABEL: st_global_f64
define void @st_global_f64(ptr addrspace(1) %ptr, double %a) {
; G32: st.global.f64 [%r{{[0-9]+}}], %fd{{[0-9]+}}
; G64: st.global.f64 [%rd{{[0-9]+}}], %fd{{[0-9]+}}
; ALL: ret
  store double %a, ptr addrspace(1) %ptr
  ret void
}
; ALL-LABEL: st_shared_f64
define void @st_shared_f64(ptr addrspace(3) %ptr, double %a) {
; LS32: st.shared.f64 [%r{{[0-9]+}}], %fd{{[0-9]+}}
; LS64: st.shared.f64 [%rd{{[0-9]+}}], %fd{{[0-9]+}}
; ALL: ret
  store double %a, ptr addrspace(3) %ptr
  ret void
}
; ALL-LABEL: st_local_f64
define void @st_local_f64(ptr addrspace(5) %ptr, double %a) {
; LS32: st.local.f64 [%r{{[0-9]+}}], %fd{{[0-9]+}}
; LS64: st.local.f64 [%rd{{[0-9]+}}], %fd{{[0-9]+}}
; ALL: ret
  store double %a, ptr addrspace(5) %ptr
  ret void
}