1; RUN: llc < %s -march=nvptx -mcpu=sm_20 -verify-machineinstrs | FileCheck %s --check-prefix=PTX32
2; RUN: llc < %s -march=nvptx64 -mcpu=sm_20 -verify-machineinstrs | FileCheck %s --check-prefix=PTX64
3; RUN: %if ptxas %{ llc < %s -march=nvptx -mcpu=sm_20 -verify-machineinstrs | %ptxas-verify %}
4; RUN: %if ptxas %{ llc < %s -march=nvptx64 -mcpu=sm_20 -verify-machineinstrs | %ptxas-verify %}
5
6; Ensure we access the local stack properly
7
8; PTX32:        mov.u32          %SPL, __local_depot{{[0-9]+}};
9; PTX32:        cvta.local.u32   %SP, %SPL;
10; PTX32:        ld.param.u32     %r{{[0-9]+}}, [foo_param_0];
11; PTX32:        st.volatile.u32  [%SP+0], %r{{[0-9]+}};
12; PTX64:        mov.u64          %SPL, __local_depot{{[0-9]+}};
13; PTX64:        cvta.local.u64   %SP, %SPL;
14; PTX64:        ld.param.u32     %r{{[0-9]+}}, [foo_param_0];
15; PTX64:        st.volatile.u32  [%SP+0], %r{{[0-9]+}};
16define void @foo(i32 %a) {
17  %local = alloca i32, align 4
18  store volatile i32 %a, i32* %local
19  ret void
20}
21
22; PTX32:        mov.u32          %SPL, __local_depot{{[0-9]+}};
23; PTX32:        cvta.local.u32   %SP, %SPL;
24; PTX32:        ld.param.u32     %r{{[0-9]+}}, [foo2_param_0];
25; PTX32:        add.u32          %r[[SP_REG:[0-9]+]], %SPL, 0;
26; PTX32:        st.local.u32  [%r[[SP_REG]]], %r{{[0-9]+}};
27; PTX64:        mov.u64          %SPL, __local_depot{{[0-9]+}};
28; PTX64:        cvta.local.u64   %SP, %SPL;
29; PTX64:        ld.param.u32     %r{{[0-9]+}}, [foo2_param_0];
30; PTX64:        add.u64          %rd[[SP_REG:[0-9]+]], %SPL, 0;
31; PTX64:        st.local.u32  [%rd[[SP_REG]]], %r{{[0-9]+}};
32define void @foo2(i32 %a) {
33  %local = alloca i32, align 4
34  store i32 %a, i32* %local
35  call void @bar(i32* %local)
36  ret void
37}
38
39declare void @bar(i32* %a)
40
41!nvvm.annotations = !{!0}
42!0 = !{void (i32)* @foo2, !"kernel", i32 1}
43
44; PTX32:        mov.u32          %SPL, __local_depot{{[0-9]+}};
45; PTX32-NOT:    cvta.local.u32   %SP, %SPL;
46; PTX32:        ld.param.u32     %r{{[0-9]+}}, [foo3_param_0];
47; PTX32:        add.u32          %r{{[0-9]+}}, %SPL, 0;
48; PTX32:        st.local.u32  [%r{{[0-9]+}}], %r{{[0-9]+}};
49; PTX64:        mov.u64          %SPL, __local_depot{{[0-9]+}};
50; PTX64-NOT:    cvta.local.u64   %SP, %SPL;
51; PTX64:        ld.param.u32     %r{{[0-9]+}}, [foo3_param_0];
52; PTX64:        add.u64          %rd{{[0-9]+}}, %SPL, 0;
53; PTX64:        st.local.u32  [%rd{{[0-9]+}}], %r{{[0-9]+}};
54define void @foo3(i32 %a) {
55  %local = alloca [3 x i32], align 4
56  %1 = bitcast [3 x i32]* %local to i32*
57  %2 = getelementptr inbounds i32, i32* %1, i32 %a
58  store i32 %a, i32* %2
59  ret void
60}
61
62; PTX32:        cvta.local.u32   %SP, %SPL;
63; PTX32:        add.u32          {{%r[0-9]+}}, %SP, 0;
64; PTX32:        add.u32          {{%r[0-9]+}}, %SPL, 0;
65; PTX32:        add.u32          {{%r[0-9]+}}, %SP, 4;
66; PTX32:        add.u32          {{%r[0-9]+}}, %SPL, 4;
67; PTX32:        st.local.u32     [{{%r[0-9]+}}], {{%r[0-9]+}}
68; PTX32:        st.local.u32     [{{%r[0-9]+}}], {{%r[0-9]+}}
69; PTX64:        cvta.local.u64   %SP, %SPL;
70; PTX64:        add.u64          {{%rd[0-9]+}}, %SP, 0;
71; PTX64:        add.u64          {{%rd[0-9]+}}, %SPL, 0;
72; PTX64:        add.u64          {{%rd[0-9]+}}, %SP, 4;
73; PTX64:        add.u64          {{%rd[0-9]+}}, %SPL, 4;
74; PTX64:        st.local.u32     [{{%rd[0-9]+}}], {{%r[0-9]+}}
75; PTX64:        st.local.u32     [{{%rd[0-9]+}}], {{%r[0-9]+}}
76define void @foo4() {
77  %A = alloca i32
78  %B = alloca i32
79  store i32 0, i32* %A
80  store i32 0, i32* %B
81  call void @bar(i32* %A)
82  call void @bar(i32* %B)
83  ret void
84}
85