xref: /llvm-project/llvm/test/CodeGen/SPIRV/function/alloca-load-store.ll (revision 67d3ef74b31e1517d4f679e754cc2b3041c95901)
1; RUN: llc -verify-machineinstrs -O0 -mtriple=spirv32-unknown-unknown %s -o - | FileCheck %s
2
3; CHECK-DAG: OpName %[[#BAR:]] "bar"
4; CHECK-DAG: OpName %[[#FOO:]] "foo"
5; CHECK-DAG: OpName %[[#GOO:]] "goo"
6
7; CHECK-DAG: %[[#INT:]] = OpTypeInt 32
8; CHECK-DAG: %[[#STACK_PTR_INT:]] = OpTypePointer Function %[[#INT]]
9; CHECK-DAG: %[[#GLOBAL_PTR_INT:]] = OpTypePointer CrossWorkgroup %[[#INT]]
10; CHECK-DAG: %[[#FN1:]] = OpTypeFunction %[[#INT]] %[[#INT]]
11; CHECK-DAG: %[[#FN2:]] = OpTypeFunction %[[#INT]] %[[#INT]] %[[#GLOBAL_PTR_INT]]
12
13define i32 @bar(i32 %a) {
14  %p = alloca i32
15  store i32 %a, i32* %p
16  %b = load i32, i32* %p
17  ret i32 %b
18}
19
20; CHECK: %[[#BAR]] = OpFunction %[[#INT]] None %[[#FN1]]
21; CHECK: %[[#A:]] = OpFunctionParameter %[[#INT]]
22; CHECK: OpLabel
23; CHECK: %[[#P:]] = OpVariable %[[#STACK_PTR_INT]] Function
24; CHECK: OpStore %[[#P]] %[[#A]]
25; CHECK: %[[#B:]] = OpLoad %[[#INT]] %[[#P]]
26; CHECK: OpReturnValue %[[#B]]
27; CHECK: OpFunctionEnd
28
29
30define i32 @foo(i32 %a) {
31  %p = alloca i32
32  store volatile i32 %a, i32* %p
33  %b = load volatile i32, i32* %p
34  ret i32 %b
35}
36
37; CHECK: %[[#FOO]] = OpFunction %[[#INT]] None %[[#FN1]]
38; CHECK: %[[#A:]] = OpFunctionParameter %[[#INT]]
39; CHECK: OpLabel
40; CHECK: %[[#P:]] = OpVariable %[[#STACK_PTR_INT]] Function
41; CHECK: OpStore %[[#P]] %[[#A]] Volatile
42; CHECK: %[[#B:]] = OpLoad %[[#INT]] %[[#P]] Volatile
43; CHECK: OpReturnValue %[[#B]]
44; CHECK: OpFunctionEnd
45
46
47;; Test load and store in global address space.
48define i32 @goo(i32 %a, ptr addrspace(1) %p) {
49  store i32 %a, i32 addrspace(1)* %p
50  %b = load i32, i32 addrspace(1)* %p
51  ret i32 %b
52}
53
54; CHECK: %[[#GOO]] = OpFunction %[[#INT]] None %[[#FN2]]
55; CHECK: %[[#A:]] = OpFunctionParameter %[[#INT]]
56; CHECK: %[[#P:]] = OpFunctionParameter %[[#GLOBAL_PTR_INT]]
57; CHECK: OpLabel
58; CHECK: OpStore %[[#P]] %[[#A]]
59; CHECK: %[[#B:]] = OpLoad %[[#INT]] %[[#P]]
60; CHECK: OpReturnValue %[[#B]]
61; CHECK: OpFunctionEnd
62