xref: /llvm-project/llvm/test/CodeGen/SPIRV/pointers/complex.ll (revision 67d3ef74b31e1517d4f679e754cc2b3041c95901)
1; RUN: llc -verify-machineinstrs -O0 -mtriple=spirv64-unknown-unknown %s -o - | FileCheck %s
2; RUN: %if spirv-tools %{ llc -O0 -mtriple=spirv64-unknown-unknown %s -o - -filetype=obj | spirv-val %}
3
4; RUN: llc -verify-machineinstrs -O0 -mtriple=spirv32-unknown-unknown %s -o - | FileCheck %s
5; RUN: %if spirv-tools %{ llc -O0 -mtriple=spirv32-unknown-unknown %s -o - -filetype=obj | spirv-val %}
6
7; CHECK-DAG: OpName %[[#Foo:]] "foo"
8; CHECK-DAG: OpName %[[#CSQRT:]] "csqrt"
9; CHECK-DAG: %[[#FloatTy:]] = OpTypeFloat 64
10; CHECK-DAG: %[[#StructTy:]] = OpTypeStruct %[[#FloatTy]] %[[#FloatTy]]
11; CHECK-DAG: %[[#GPtrStructTy:]] = OpTypePointer Generic %[[#StructTy]]
12; CHECK-DAG: %[[#FPtrStructTy:]] = OpTypePointer Function %[[#StructTy]]
13; CHECK-DAG: %[[#WrapTy:]] = OpTypeStruct %[[#StructTy]]
14; CHECK-DAG: %[[#Int64Ty:]] = OpTypeInt 64 0
15; CHECK-DAG: %[[#Int32Ty:]] = OpTypeInt 32 0
16; CHECK-DAG: %[[#C1:]] = OpConstant %[[#Int32Ty]] 1
17; CHECK-DAG: %[[#ArrayTy:]] = OpTypeArray %[[#Int64Ty]] %[[#C1]]
18; CHECK-DAG: %[[#SArrayTy:]] = OpTypeStruct %[[#ArrayTy]]
19; CHECK-DAG: %[[#SSArrayTy:]] = OpTypeStruct %[[#SArrayTy]]
20; CHECK-DAG: %[[#CWPtrComplexTy:]] = OpTypePointer CrossWorkgroup %[[#WrapTy]]
21; CHECK-DAG: %[[#IdTy:]] = OpTypePointer Function %[[#SSArrayTy]]
22; CHECK: %[[#Foo]] = OpFunction
23; CHECK: OpFunctionParameter %[[#CWPtrComplexTy]]
24; CHECK: OpFunctionParameter %[[#IdTy]]
25; CHECK: %[[#CSQRT]] = OpFunction
26; CHECK: OpFunctionParameter %[[#GPtrStructTy]]
27; CHECK: OpFunctionParameter %[[#FPtrStructTy]]
28
29%"class.id" = type { %"class.array" }
30%"class.array" = type { [1 x i64] }
31%"class.complex" = type { { double, double } }
32
33define weak_odr dso_local spir_kernel void @foo(ptr addrspace(1) align 8 %_arg_buf_out1_access, ptr byval(%"class.id") align 8 %_arg_buf_out1_access3) {
34entry:
35  %tmp.i.i = alloca { double, double }, align 8
36  %byval-temp.i.i = alloca { double, double }, align 8
37  %idxvalue = load i64, ptr %_arg_buf_out1_access3, align 8
38  %add.ptr.i = getelementptr inbounds %"class.complex", ptr addrspace(1) %_arg_buf_out1_access, i64 %idxvalue
39  %tmp.ascast.i.i = addrspacecast ptr %tmp.i.i to ptr addrspace(4)
40  %byval-temp.imagp.i.i = getelementptr inbounds i8, ptr %byval-temp.i.i, i64 8
41  store double -1.000000e+00, ptr %byval-temp.i.i, align 8
42  store double 0.000000e+00, ptr %byval-temp.imagp.i.i, align 8
43  call spir_func void @csqrt(ptr addrspace(4) dead_on_unwind writable sret({ double, double }) align 8 %tmp.ascast.i.i, ptr nonnull byval({ double, double }) align 8 %byval-temp.i.i)
44  %tmp.ascast.real.i.i = load double, ptr %tmp.i.i, align 8
45  %tmp.ascast.imagp.i.i = getelementptr inbounds i8, ptr %tmp.i.i, i64 8
46  %tmp.ascast.imag.i.i = load double, ptr %tmp.ascast.imagp.i.i, align 8
47  store double %tmp.ascast.real.i.i, ptr addrspace(1) %add.ptr.i, align 8
48  %dest = getelementptr inbounds i8, ptr addrspace(1) %add.ptr.i, i64 8
49  store double %tmp.ascast.imag.i.i, ptr addrspace(1) %dest, align 8
50  ret void
51}
52
53define weak dso_local spir_func void @csqrt(ptr addrspace(4) dead_on_unwind noalias writable sret({ double, double }) align 8 %agg.result, ptr byval({ double, double }) align 8 %z) {
54entry:
55  %tmp = alloca { double, double }, align 8
56  %byval-temp = alloca { double, double }, align 8
57  %tmp.ascast = addrspacecast ptr %tmp to ptr addrspace(4)
58  %z.ascast.real = load double, ptr %z, align 8
59  %z.ascast.imagp = getelementptr inbounds i8, ptr %z, i64 8
60  %z.ascast.imag = load double, ptr %z.ascast.imagp, align 8
61  %byval-temp.imagp = getelementptr inbounds i8, ptr %byval-temp, i64 8
62  store double %z.ascast.real, ptr %byval-temp, align 8
63  store double %z.ascast.imag, ptr %byval-temp.imagp, align 8
64  call spir_func void @__devicelib_csqrt(ptr addrspace(4) dead_on_unwind writable sret({ double, double }) align 8 %tmp.ascast, ptr nonnull byval({ double, double }) align 8 %byval-temp) #7
65  %tmp.ascast.real = load double, ptr %tmp, align 8
66  %tmp.ascast.imagp = getelementptr inbounds i8, ptr %tmp, i64 8
67  %tmp.ascast.imag = load double, ptr %tmp.ascast.imagp, align 8
68  %agg.result.imagp = getelementptr inbounds i8, ptr addrspace(4) %agg.result, i64 8
69  store double %tmp.ascast.real, ptr addrspace(4) %agg.result, align 8
70  store double %tmp.ascast.imag, ptr addrspace(4) %agg.result.imagp, align 8
71  ret void
72}
73
74declare extern_weak dso_local spir_func void @__devicelib_csqrt(ptr addrspace(4) dead_on_unwind writable sret({ double, double }) align 8, ptr byval({ double, double }) align 8) local_unnamed_addr #3
75