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