summaryrefslogtreecommitdiff
path: root/test/CodeGen/NVPTX/local-stack-frame.ll
diff options
context:
space:
mode:
authorDimitry Andric <dim@FreeBSD.org>2015-07-05 14:21:36 +0000
committerDimitry Andric <dim@FreeBSD.org>2015-07-05 14:21:36 +0000
commit1a82d4c088707c791c792f6822f611b47a12bdfe (patch)
tree7c411f9b5d807f7f204fdd16965d8925a82b6d18 /test/CodeGen/NVPTX/local-stack-frame.ll
parent3a0822f094b578157263e04114075ad7df81db41 (diff)
Notes
Diffstat (limited to 'test/CodeGen/NVPTX/local-stack-frame.ll')
-rw-r--r--test/CodeGen/NVPTX/local-stack-frame.ll72
1 files changed, 68 insertions, 4 deletions
diff --git a/test/CodeGen/NVPTX/local-stack-frame.ll b/test/CodeGen/NVPTX/local-stack-frame.ll
index 377eee9170e6..ef1b7da6ad0f 100644
--- a/test/CodeGen/NVPTX/local-stack-frame.ll
+++ b/test/CodeGen/NVPTX/local-stack-frame.ll
@@ -3,12 +3,12 @@
; Ensure we access the local stack properly
-; PTX32: mov.u32 %r{{[0-9]+}}, __local_depot{{[0-9]+}};
-; PTX32: cvta.local.u32 %SP, %r{{[0-9]+}};
+; PTX32: mov.u32 %SPL, __local_depot{{[0-9]+}};
+; PTX32: cvta.local.u32 %SP, %SPL;
; PTX32: ld.param.u32 %r{{[0-9]+}}, [foo_param_0];
; PTX32: st.volatile.u32 [%SP+0], %r{{[0-9]+}};
-; PTX64: mov.u64 %rd{{[0-9]+}}, __local_depot{{[0-9]+}};
-; PTX64: cvta.local.u64 %SP, %rd{{[0-9]+}};
+; PTX64: mov.u64 %SPL, __local_depot{{[0-9]+}};
+; PTX64: cvta.local.u64 %SP, %SPL;
; PTX64: ld.param.u32 %r{{[0-9]+}}, [foo_param_0];
; PTX64: st.volatile.u32 [%SP+0], %r{{[0-9]+}};
define void @foo(i32 %a) {
@@ -16,3 +16,67 @@ define void @foo(i32 %a) {
store volatile i32 %a, i32* %local
ret void
}
+
+; PTX32: mov.u32 %SPL, __local_depot{{[0-9]+}};
+; PTX32: cvta.local.u32 %SP, %SPL;
+; PTX32: ld.param.u32 %r{{[0-9]+}}, [foo2_param_0];
+; PTX32: add.u32 %r[[SP_REG:[0-9]+]], %SPL, 0;
+; PTX32: st.local.u32 [%r[[SP_REG]]], %r{{[0-9]+}};
+; PTX64: mov.u64 %SPL, __local_depot{{[0-9]+}};
+; PTX64: cvta.local.u64 %SP, %SPL;
+; PTX64: ld.param.u32 %r{{[0-9]+}}, [foo2_param_0];
+; PTX64: add.u64 %rd[[SP_REG:[0-9]+]], %SPL, 0;
+; PTX64: st.local.u32 [%rd[[SP_REG]]], %r{{[0-9]+}};
+define void @foo2(i32 %a) {
+ %local = alloca i32, align 4
+ store i32 %a, i32* %local
+ call void @bar(i32* %local)
+ ret void
+}
+
+declare void @bar(i32* %a)
+
+!nvvm.annotations = !{!0}
+!0 = !{void (i32)* @foo2, !"kernel", i32 1}
+
+; PTX32: mov.u32 %SPL, __local_depot{{[0-9]+}};
+; PTX32-NOT: cvta.local.u32 %SP, %SPL;
+; PTX32: ld.param.u32 %r{{[0-9]+}}, [foo3_param_0];
+; PTX32: add.u32 %r{{[0-9]+}}, %SPL, 0;
+; PTX32: st.local.u32 [%r{{[0-9]+}}], %r{{[0-9]+}};
+; PTX64: mov.u64 %SPL, __local_depot{{[0-9]+}};
+; PTX64-NOT: cvta.local.u64 %SP, %SPL;
+; PTX64: ld.param.u32 %r{{[0-9]+}}, [foo3_param_0];
+; PTX64: add.u64 %rd{{[0-9]+}}, %SPL, 0;
+; PTX64: st.local.u32 [%rd{{[0-9]+}}], %r{{[0-9]+}};
+define void @foo3(i32 %a) {
+ %local = alloca [3 x i32], align 4
+ %1 = bitcast [3 x i32]* %local to i32*
+ %2 = getelementptr inbounds i32, i32* %1, i32 %a
+ store i32 %a, i32* %2
+ ret void
+}
+
+; PTX32: cvta.local.u32 %SP, %SPL;
+; PTX32: add.u32 {{%r[0-9]+}}, %SP, 0;
+; PTX32: add.u32 {{%r[0-9]+}}, %SPL, 0;
+; PTX32: add.u32 {{%r[0-9]+}}, %SP, 4;
+; PTX32: add.u32 {{%r[0-9]+}}, %SPL, 4;
+; PTX32: st.local.u32 [{{%r[0-9]+}}], {{%r[0-9]+}}
+; PTX32: st.local.u32 [{{%r[0-9]+}}], {{%r[0-9]+}}
+; PTX64: cvta.local.u64 %SP, %SPL;
+; PTX64: add.u64 {{%rd[0-9]+}}, %SP, 0;
+; PTX64: add.u64 {{%rd[0-9]+}}, %SPL, 0;
+; PTX64: add.u64 {{%rd[0-9]+}}, %SP, 4;
+; PTX64: add.u64 {{%rd[0-9]+}}, %SPL, 4;
+; PTX64: st.local.u32 [{{%rd[0-9]+}}], {{%r[0-9]+}}
+; PTX64: st.local.u32 [{{%rd[0-9]+}}], {{%r[0-9]+}}
+define void @foo4() {
+ %A = alloca i32
+ %B = alloca i32
+ store i32 0, i32* %A
+ store i32 0, i32* %B
+ call void @bar(i32* %A)
+ call void @bar(i32* %B)
+ ret void
+}