1 files changed, 66 insertions, 0 deletions
diff --git a/test/CodeGen/NVPTX/call-with-alloca-buffer.ll b/test/CodeGen/NVPTX/call-with-alloca-buffer.ll
new file mode 100644
index 0000000..28dfa46
--- /dev/null
+++ b/test/CodeGen/NVPTX/call-with-alloca-buffer.ll
@@ -0,0 +1,66 @@
+; RUN: llc < %s -march=nvptx64 -mcpu=sm_20 | FileCheck %s
+
+; Checks how NVPTX lowers alloca buffers and their passing to functions.
+;
+; Produced with the following CUDA code:
+;  extern "C" __attribute__((device)) void callee(float* f, char* buf);
+;
+;  extern "C" __attribute__((global)) void kernel_func(float* a) {
+;    char buf[4 * sizeof(float)];
+;    *(reinterpret_cast<float*>(&buf[0])) = a[0];
+;    *(reinterpret_cast<float*>(&buf[1])) = a[1];
+;    *(reinterpret_cast<float*>(&buf[2])) = a[2];
+;    *(reinterpret_cast<float*>(&buf[3])) = a[3];
+;    callee(a, buf);
+;  }
+
+; CHECK: .visible .entry kernel_func
+define void @kernel_func(float* %a) {
+entry:
+  %buf = alloca [16 x i8], align 4
+
+; CHECK: .local .align 4 .b8 	__local_depot0[16]
+; CHECK: mov.u64 %rl[[BUF_REG:[0-9]+]]
+; CHECK: cvta.local.u64 %SP, %rl[[BUF_REG]]
+
+; CHECK: ld.param.u64 %rl[[A_REG:[0-9]+]], [kernel_func_param_0]
+; CHECK: ld.f32 %f[[A0_REG:[0-9]+]], [%rl[[A_REG]]]
+; CHECK: st.f32 [%SP+0], %f[[A0_REG]]
+
+  %0 = load float* %a, align 4
+  %1 = bitcast [16 x i8]* %buf to float*
+  store float %0, float* %1, align 4
+  %arrayidx2 = getelementptr inbounds float* %a, i64 1
+  %2 = load float* %arrayidx2, align 4
+  %arrayidx3 = getelementptr inbounds [16 x i8]* %buf, i64 0, i64 1
+  %3 = bitcast i8* %arrayidx3 to float*
+  store float %2, float* %3, align 4
+  %arrayidx4 = getelementptr inbounds float* %a, i64 2
+  %4 = load float* %arrayidx4, align 4
+  %arrayidx5 = getelementptr inbounds [16 x i8]* %buf, i64 0, i64 2
+  %5 = bitcast i8* %arrayidx5 to float*
+  store float %4, float* %5, align 4
+  %arrayidx6 = getelementptr inbounds float* %a, i64 3
+  %6 = load float* %arrayidx6, align 4
+  %arrayidx7 = getelementptr inbounds [16 x i8]* %buf, i64 0, i64 3
+  %7 = bitcast i8* %arrayidx7 to float*
+  store float %6, float* %7, align 4
+
+; CHECK: add.u64 %rl[[SP_REG:[0-9]+]], %SP, 0
+; CHECK:        .param .b64 param0;
+; CHECK-NEXT:   st.param.b64  [param0+0], %rl[[A_REG]]
+; CHECK-NEXT:   .param .b64 param1;
+; CHECK-NEXT:   st.param.b64  [param1+0], %rl[[SP_REG]]
+; CHECK-NEXT:   call.uni
+; CHECK-NEXT:   callee,
+
+  %arraydecay = getelementptr inbounds [16 x i8]* %buf, i64 0, i64 0
+  call void @callee(float* %a, i8* %arraydecay) #2
+  ret void
+}
+
+declare void @callee(float*, i8*)
+
+!nvvm.annotations = !{!0}
+
+!0 = metadata !{void (float*)* @kernel_func, metadata !"kernel", i32 1}