• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1; RUN: llc < %s -march=nvptx64 -mcpu=sm_20 | FileCheck %s
2
3target datalayout = "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-f32:32:32-f64:64:64-v16:16:16-v32:32:32-v64:64:64-v128:128:128-n16:32:64"
4target triple = "nvptx64-nvidia-cuda"
5
6; Verify that both %input and %output are converted to global pointers and then
7; addrspacecast'ed back to the original type.
8define void @kernel(float* %input, float* %output) {
9; CHECK-LABEL: .visible .entry kernel(
10; CHECK: cvta.to.global.u64
11; CHECK: cvta.to.global.u64
12  %1 = load float, float* %input, align 4
13; CHECK: ld.global.f32
14  store float %1, float* %output, align 4
15; CHECK: st.global.f32
16  ret void
17}
18
19define void @kernel2(float addrspace(1)* %input, float addrspace(1)* %output) {
20; CHECK-LABEL: .visible .entry kernel2(
21; CHECK-NOT: cvta.to.global.u64
22  %1 = load float, float addrspace(1)* %input, align 4
23; CHECK: ld.global.f32
24  store float %1, float addrspace(1)* %output, align 4
25; CHECK: st.global.f32
26  ret void
27}
28
29%struct.S = type { i32*, i32* }
30
31define void @ptr_in_byval(%struct.S* byval %input, i32* %output) {
32; CHECK-LABEL: .visible .entry ptr_in_byval(
33; CHECK: cvta.to.global.u64
34; CHECK: cvta.to.global.u64
35  %b_ptr = getelementptr inbounds %struct.S, %struct.S* %input, i64 0, i32 1
36  %b = load i32*, i32** %b_ptr, align 4
37  %v = load i32, i32* %b, align 4
38; CHECK: ld.global.u32
39  store i32 %v, i32* %output, align 4
40; CHECK: st.global.u32
41  ret void
42}
43
44!nvvm.annotations = !{!0, !1, !2}
45!0 = !{void (float*, float*)* @kernel, !"kernel", i32 1}
46!1 = !{void (float addrspace(1)*, float addrspace(1)*)* @kernel2, !"kernel", i32 1}
47!2 = !{void (%struct.S*, i32*)* @ptr_in_byval, !"kernel", i32 1}
48